Fix: push_to_hub to use correct model (#98)

fixie-ai · Aug 27, 2024 · 8b6a3aa · 8b6a3aa
1 parent 732b7ba
commit 8b6a3aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ultravox/tools/push_to_hub.py b/ultravox/tools/push_to_hub.py
@@ -47,7 +47,7 @@ def main(args: UploadToHubArgs):
 
     print("Model uploaded. Testing model...")
     loaded_pipe = transformers.pipeline(
-        model="fixie-ai/ultravox-v0_2", trust_remote_code=True
+        model=args.hf_upload_model, trust_remote_code=True
     )
     ds = datasets.BoolQDataset(datasets.VoiceDatasetArgs())
     sample = next(iter(ds))