diff --git a/apps/models_provider/impl/xinference_model_provider/credential/stt.py b/apps/models_provider/impl/xinference_model_provider/credential/stt.py index 3cdbb53f254..e4a389fd476 100644 --- a/apps/models_provider/impl/xinference_model_provider/credential/stt.py +++ b/apps/models_provider/impl/xinference_model_provider/credential/stt.py @@ -27,7 +27,7 @@ def is_valid(self, model_type: str, model_name, model_credential: Dict[str, obje else: return False try: - model = provider.get_model(model_type, model_name, model_credential) + model = provider.get_model(model_type, model_name, model_credential, **model_params) model.check_auth() except Exception as e: if isinstance(e, AppApiException): diff --git a/apps/models_provider/impl/xinference_model_provider/model/stt.py b/apps/models_provider/impl/xinference_model_provider/model/stt.py index e614d42d4d7..f8935045d1c 100644 --- a/apps/models_provider/impl/xinference_model_provider/model/stt.py +++ b/apps/models_provider/impl/xinference_model_provider/model/stt.py @@ -17,6 +17,7 @@ class XInferenceSpeechToText(MaxKBBaseModel, BaseSpeechToText): api_base: str api_key: str model: str + params: dict def __init__(self, **kwargs): super().__init__(**kwargs)