From f779fbd5cc74e6abab3dcee00090454238395d4c Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 19 Mar 2023 23:30:31 -0500 Subject: [PATCH] fix(api): remove prefix before downloading for HF hub --- api/onnx_web/convert/__main__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py index 6b63b05a..ad89ead7 100644 --- 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",