fix(api): remove prefix before downloading for HF hub

ssube · Mar 20, 2023 · f779fbd · f779fbd
1 parent 1971226
commit f779fbd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py
@@ -181,7 +181,7 @@ def fetch_model(
         # from_pretrained has a bunch of useful logic that snapshot_download by itself down not
         if hf_hub_fetch:
             return hf_hub_download(
-                repo_id=source,
+                repo_id=hub_source,
                 filename=hf_hub_filename,
                 cache_dir=cache_path,
                 force_filename=f"{name}.bin",