feat:add fishaudio in xinference (#8100)

2 years ago · d28446301f
parent 9050f92e5b
commit d28446301f
1 changed files with 1 additions and 1 deletions
--- 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')