Spaces:

Felix5572
/

ChuanhuChatGPT

Sleeping

Tuchuanhuhuhu commited on Dec 25, 2023

Commit

180ee81

1 Parent(s): d0c0cec

bugfix: 修复Qwen bug

Files changed (1) hide show

modules/models/base_model.py CHANGED Viewed

@@ -221,9 +221,9 @@ class BaseLLMModel:
     ) -> None:
         self.history = []
         self.all_token_counts = []
-        if model_name in MODEL_METADATA:
             self.model_name = MODEL_METADATA[model_name]["model_name"]
-        else:
             self.model_name = model_name
         self.model_type = ModelType.get_type(model_name)
         try:

     ) -> None:
         self.history = []
         self.all_token_counts = []
+        try:
             self.model_name = MODEL_METADATA[model_name]["model_name"]
+        except:
             self.model_name = model_name
         self.model_type = ModelType.get_type(model_name)
         try: