Merge pull request #1486 from CerebriumAI/feature/ultravox

Feature/ultravox - bug fixes
This commit is contained in:
milo157
2025-04-01 00:03:26 +02:00
committed by GitHub
parent 2fb9aa4d76
commit ed387e876a
4 changed files with 19 additions and 13 deletions

View File

@@ -34,7 +34,7 @@ logger.add(sys.stderr, level="DEBUG")
# Want to initialize the ultravox processor since it takes time to load the model and dont
# want to load it every time the pipeline is run
ultravox_processor = UltravoxSTTService(
model_size="fixie-ai/ultravox-v0_4_1-llama-3_1-8b",
model_name="fixie-ai/ultravox-v0_5-llama-3_1-8b",
hf_token=os.getenv("HF_TOKEN"),
)