fix(api): remove prefix before downloading for HF hub
This commit is contained in:
parent
19712262e6
commit
f779fbd5cc
|
@ -181,7 +181,7 @@ def fetch_model(
|
||||||
# from_pretrained has a bunch of useful logic that snapshot_download by itself down not
|
# from_pretrained has a bunch of useful logic that snapshot_download by itself down not
|
||||||
if hf_hub_fetch:
|
if hf_hub_fetch:
|
||||||
return hf_hub_download(
|
return hf_hub_download(
|
||||||
repo_id=source,
|
repo_id=hub_source,
|
||||||
filename=hf_hub_filename,
|
filename=hf_hub_filename,
|
||||||
cache_dir=cache_path,
|
cache_dir=cache_path,
|
||||||
force_filename=f"{name}.bin",
|
force_filename=f"{name}.bin",
|
||||||
|
|
Loading…
Reference in New Issue