fix enabling sparkv4 (#1936)

binary-husky · Aug 12, 2024 · a119ab3 · a119ab3
1 parent f9384e4
commit a119ab3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/request_llms/bridge_all.py b/request_llms/bridge_all.py
@@ -881,7 +881,7 @@ def decode(self, *args, **kwargs):
         })
     except:
         print(trimmed_format_exc())
-if "sparkv3" in AVAIL_LLM_MODELS or "sparkv3.5" in AVAIL_LLM_MODELS:   # 讯飞星火认知大模型
+if any(x in AVAIL_LLM_MODELS for x in ("sparkv3", "sparkv3.5", "sparkv4")):   # 讯飞星火认知大模型
     try:
         from .bridge_spark import predict_no_ui_long_connection as spark_noui
         from .bridge_spark import predict as spark_ui