From 483b7dd291c8bd3014f5545e2d96efcd16f9a97f Mon Sep 17 00:00:00 2001 From: imClumsyPanda Date: Sun, 21 May 2023 22:41:22 +0800 Subject: [PATCH] update model_config.py --- configs/model_config.py | 13 +------------ 1 file changed, 1 insertion(+), 12 deletions(-) diff --git a/configs/model_config.py b/configs/model_config.py index 7658b2a..ab97b97 100644 --- a/configs/model_config.py +++ b/configs/model_config.py @@ -52,18 +52,7 @@ llm_model_dict = { "local_model_path": None, "provides": "ChatGLM" }, - "llama-7b-hf": { - "name": "llama-7b-hf", - "pretrained_model_name": "llama-7b-hf", - "local_model_path": None, - "provides": "LLamaLLM" - }, - "vicuna-13b-hf": { - "name": "vicuna-13b-hf", - "pretrained_model_name": "vicuna-13b-hf", - "local_model_path": None, - "provides": "LLamaLLM" - }, + "chatyuan": { "name": "chatyuan", "pretrained_model_name": "ClueAI/ChatYuan-large-v2",