Update model_config.py

eosphoros-ai · Jun 25, 2023 · aa4115e · aa4115e
1 parent c19a7b1
commit aa4115e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pilot/configs/model_config.py b/pilot/configs/model_config.py
@@ -40,6 +40,7 @@
     "chatglm-6b-int4": os.path.join(MODEL_PATH, "chatglm-6b-int4"),
     "chatglm-6b": os.path.join(MODEL_PATH, "chatglm-6b"),
     "chatglm2-6b": os.path.join(MODEL_PATH, "chatglm2-6b"),
+    "chatglm2-6b-int4": os.path.join(MODEL_PATH, "chatglm2-6b-int4"),
     "text2vec-base": os.path.join(MODEL_PATH, "text2vec-base-chinese"),
     "guanaco-33b-merged": os.path.join(MODEL_PATH, "guanaco-33b-merged"),
     "falcon-40b": os.path.join(MODEL_PATH, "falcon-40b"),