"tests/python/pytorch/dataloading/test_dataloader.py" did not exist on "d70beba762c3b151edb8578f1dfbdce01c0dfa73"
Unverified Commit 6198f8a1 authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

[Modular] Allow ModularPipeline to load from revisions (#12592)



* update

* update

* update

* update

---------
Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent dcfb18a2
...@@ -358,6 +358,7 @@ def get_cached_module_file( ...@@ -358,6 +358,7 @@ def get_cached_module_file(
proxies=proxies, proxies=proxies,
local_files_only=local_files_only, local_files_only=local_files_only,
local_dir=local_dir, local_dir=local_dir,
revision=revision,
token=token, token=token,
) )
submodule = os.path.join("local", "--".join(pretrained_model_name_or_path.split("/"))) submodule = os.path.join("local", "--".join(pretrained_model_name_or_path.split("/")))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment