update model_config.py

2023-05-21 22:41:22 +08:00 · 2023-05-21 22:41:22 +08:00 · 483b7dd291
parent 12316d0658
commit 483b7dd291
1 changed files with 1 additions and 12 deletions
--- 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",