feat:add fishaudio in xinference (#8100)

This commit is contained in:
呆萌闷油瓶
2024-09-08 23:58:02 +08:00
committed by GitHub
parent 9050f92e5b
commit d28446301f

View File

@@ -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')