Bladeren bron

feat:add fishaudio in xinference (#8100)

呆萌闷油瓶 7 maanden geleden
bovenliggende
commit
d28446301f
1 gewijzigde bestanden met toevoegingen van 1 en 1 verwijderingen
  1. 1 1
      api/core/model_runtime/model_providers/xinference/xinference_helper.py

+ 1 - 1
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')