From 73ae1bfb846b1c8f067e640a7af1e394fc706619 Mon Sep 17 00:00:00 2001 From: Farzad Abdolhosseini Date: Fri, 23 Aug 2024 09:46:46 -0700 Subject: [PATCH] Fix: push_to_hub to use correct model --- 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))