refactor: qwq model
This commit is contained in:
parent
bdaeb1bec4
commit
2fe4248785
@ -14,7 +14,7 @@ class BaiLianChatModel(MaxKBBaseModel, BaseChatOpenAI):
|
|||||||
@staticmethod
|
@staticmethod
|
||||||
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
|
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
|
||||||
optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)
|
optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)
|
||||||
if 'qwen-omni-turbo' in model_name or 'qwq-plus' in model_name:
|
if 'qwen-omni-turbo' in model_name or 'qwq' in model_name:
|
||||||
optional_params['streaming'] = True
|
optional_params['streaming'] = True
|
||||||
return BaiLianChatModel(
|
return BaiLianChatModel(
|
||||||
model=model_name,
|
model=model_name,
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user