Support TTS and Speech2Text for Model Provider GPUStack (#12381)

This commit is contained in:
Alex Chen
2025-01-07 09:42:11 +08:00
committed by GitHub
parent 409cc7d9b0
commit 2bb521b135
9 changed files with 216 additions and 14 deletions

View File

@@ -0,0 +1,24 @@
import os
from core.model_runtime.model_providers.gpustack.tts.tts import GPUStackText2SpeechModel
def test_invoke_model():
model = GPUStackText2SpeechModel()
result = model.invoke(
model="cosyvoice-300m-sft",
tenant_id="test",
credentials={
"endpoint_url": os.environ.get("GPUSTACK_SERVER_URL"),
"api_key": os.environ.get("GPUSTACK_API_KEY"),
},
content_text="Hello world",
voice="Chinese Female",
)
content = b""
for chunk in result:
content += chunk
assert content != b""