diff --git a/api/core/model_runtime/model_providers/xinference/xinference_helper.py b/api/core/model_runtime/model_providers/xinference/xinference_helper.py index 75161ad37..151166f16 100644 --- a/api/core/model_runtime/model_providers/xinference/xinference_helper.py +++ b/api/core/model_runtime/model_providers/xinference/xinference_helper.py @@ -89,7 +89,7 @@ class XinferenceHelper: model_handle_type = 'embedding' elif response_json.get('model_type') == 'audio': model_handle_type = 'audio' - if model_family and model_family in ['ChatTTS', 'CosyVoice']: + if model_family and model_family in ['ChatTTS', 'CosyVoice', 'FishAudio']: model_ability.append('text-to-audio') else: model_ability.append('audio-to-text')