Unverified Commit 79b118e8 authored by Robin's avatar Robin Committed by GitHub
Browse files

[Fix] when run load pretain with local_files_only, local variable...


[Fix] when run load pretain with local_files_only, local variable 'cached_folder' referenced before assignment (#9376)

Fix local variable 'cached_folder' referenced before assignment in hub_utils.py

Fix when use `local_files_only=True` with `subfolder`, local variable 'cached_folder' referenced before assignment issue.
Co-authored-by: default avatarYiYi Xu <yixu310@gmail.com>
parent 9d061618
......@@ -496,7 +496,7 @@ def _get_checkpoint_shard_files(
local_dir=cache_dir, subfolder=subfolder, original_shard_filenames=original_shard_filenames
)
if subfolder is not None:
cached_folder = os.path.join(cached_folder, subfolder)
cached_folder = os.path.join(cache_dir, subfolder)
return cached_folder, sharded_metadata
......
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