Spaces:
Sleeping
Sleeping
调试qianfan成功,支持qianfan所有模型
Browse files- modules/models/base_model.py +1 -1
- modules/presets.py +26 -9
modules/models/base_model.py
CHANGED
|
@@ -194,7 +194,7 @@ class ModelType(Enum):
|
|
| 194 |
model_type = ModelType.Claude
|
| 195 |
elif "qwen" in model_name_lower:
|
| 196 |
model_type = ModelType.Qwen
|
| 197 |
-
elif "ernie" in model_name_lower:
|
| 198 |
model_type = ModelType.ERNIE
|
| 199 |
elif "dall" in model_name_lower:
|
| 200 |
model_type = ModelType.DALLE3
|
|
|
|
| 194 |
model_type = ModelType.Claude
|
| 195 |
elif "qwen" in model_name_lower:
|
| 196 |
model_type = ModelType.Qwen
|
| 197 |
+
elif "ernie" in model_name_lower or "xuanyuan" in model_name_lower:
|
| 198 |
model_type = ModelType.ERNIE
|
| 199 |
elif "dall" in model_name_lower:
|
| 200 |
model_type = ModelType.DALLE3
|
modules/presets.py
CHANGED
|
@@ -84,9 +84,22 @@ ONLINE_MODELS = [
|
|
| 84 |
"讯飞星火大模型V2.0",
|
| 85 |
"讯飞星火大模型V1.5",
|
| 86 |
"Claude",
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 90 |
]
|
| 91 |
|
| 92 |
LOCAL_MODELS = [
|
|
@@ -164,16 +177,20 @@ MODEL_METADATA = {
|
|
| 164 |
"model_name": "Claude",
|
| 165 |
"token_limit": 4096,
|
| 166 |
},
|
| 167 |
-
"
|
| 168 |
-
"model_name": "
|
| 169 |
"token_limit": 3000,
|
| 170 |
},
|
| 171 |
-
"
|
| 172 |
-
"model_name": "
|
| 173 |
"token_limit": 3000,
|
| 174 |
},
|
| 175 |
-
"
|
| 176 |
-
"model_name": "
|
|
|
|
|
|
|
|
|
|
|
|
|
| 177 |
"token_limit": 3000,
|
| 178 |
},
|
| 179 |
}
|
|
|
|
| 84 |
"讯飞星火大模型V2.0",
|
| 85 |
"讯飞星火大模型V1.5",
|
| 86 |
"Claude",
|
| 87 |
+
'XuanYuan-70B-Chat-'
|
| 88 |
+
'Qianfan-BLOOMZ-7B-compressed',
|
| 89 |
+
'ChatGLM2-6B-32K',
|
| 90 |
+
'Llama-2-7b-chat',
|
| 91 |
+
'Llama-2-70b-chat',
|
| 92 |
+
'Qianfan-Chinese-Llama-2-7B',
|
| 93 |
+
'ERNIE-Bot-8k',
|
| 94 |
+
'ERNIE-Bot-turbo',
|
| 95 |
+
'ERNIE-Bot',
|
| 96 |
+
'ERNIE-Bot-turbo-AI',
|
| 97 |
+
'AquilaChat-7B',
|
| 98 |
+
'Qianfan-Chinese-Llama-2-13B',
|
| 99 |
+
'ERNIE-Bot-4',
|
| 100 |
+
'Llama-2-13b-chat',
|
| 101 |
+
'XuanYuan-70B-Chat-4bit',
|
| 102 |
+
'BLOOMZ-7B',
|
| 103 |
]
|
| 104 |
|
| 105 |
LOCAL_MODELS = [
|
|
|
|
| 177 |
"model_name": "Claude",
|
| 178 |
"token_limit": 4096,
|
| 179 |
},
|
| 180 |
+
"ERNIE-Bot-turbo": {
|
| 181 |
+
"model_name": "ERNIE-Bot-turbo",
|
| 182 |
"token_limit": 3000,
|
| 183 |
},
|
| 184 |
+
"ERNIE-Bot": {
|
| 185 |
+
"model_name": "ERNIE-Bot",
|
| 186 |
"token_limit": 3000,
|
| 187 |
},
|
| 188 |
+
"ERNIE-Bot-4": {
|
| 189 |
+
"model_name": "ERNIE-Bot-4",
|
| 190 |
+
"token_limit": 3000,
|
| 191 |
+
},
|
| 192 |
+
"XuanYuan-70B-Chat-4bit": {
|
| 193 |
+
"model_name": "XuanYuan-70B-Chat-4bit",
|
| 194 |
"token_limit": 3000,
|
| 195 |
},
|
| 196 |
}
|