From a119ab36feb679b4a06769e07cc8169e081eac3f Mon Sep 17 00:00:00 2001 From: moetayuko Date: Mon, 12 Aug 2024 21:45:08 +0800 Subject: [PATCH] fix enabling sparkv4 (#1936) --- request_llms/bridge_all.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/request_llms/bridge_all.py b/request_llms/bridge_all.py index f08aaf76..6de07de0 100644 --- a/request_llms/bridge_all.py +++ b/request_llms/bridge_all.py @@ -881,7 +881,7 @@ if "sparkv2" in AVAIL_LLM_MODELS: # 讯飞星火认知大模型 }) 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