From 8b6a3aa4b9d3656cd76ae1ed466b5675b4d21288 Mon Sep 17 00:00:00 2001 From: Farzad Abdolhosseini Date: Tue, 27 Aug 2024 13:33:59 -0700 Subject: [PATCH] Fix: push_to_hub to use correct model (#98) --- ultravox/tools/push_to_hub.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultravox/tools/push_to_hub.py b/ultravox/tools/push_to_hub.py index d9719bbc..01945321 100644 --- 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))