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 3cdbb53f2..e4a389fd4 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 @@ class XInferenceSTTModelCredential(BaseForm, BaseModelCredential): 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 e614d42d4..f8935045d 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)