Update app.py

ypwhs · Apr 1, 2023 · 20d71c6 · 20d71c6
1 parent da43e52
commit 20d71c6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/app.py b/app.py
@@ -8,10 +8,10 @@
 print('Done'.center(64, '-'))
 
 # 加载模型
-# model_name = 'THUDM/chatglm-6b'
-# model_name = 'BelleGroup/BELLE-LLAMA-7B-2M'
+model_name = 'THUDM/chatglm-6b'
 # model_name = 'silver/chatglm-6b-int4-slim'
-model_name = 'BelleGroup/BELLE-LLAMA-7B-2M-gptq'
+# model_name = 'BelleGroup/BELLE-LLAMA-7B-2M'
+# model_name = 'BelleGroup/BELLE-LLAMA-7B-2M-gptq'
 
 if 'chatglm' in model_name.lower():
     from predictors.chatglm import ChatGLM