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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

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