소스 검색

feat:add fishaudio in xinference (#8100)

呆萌闷油瓶 7 달 전
부모
커밋
d28446301f
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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')