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

2023-03-19 23:30:31 -05:00 · 2023-03-19 23:30:31 -05:00 · f779fbd5cc
parent 19712262e6
commit f779fbd5cc
1 changed files with 1 additions and 1 deletions
--- 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",