Unverified Commit 1c73d85b authored by Aaron V's avatar Aaron V Committed by GitHub
Browse files

Update comment in modeling_utils.py (#31299)

parent 9f863d9a
...@@ -3570,7 +3570,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix ...@@ -3570,7 +3570,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
# We'll need to download and cache each checkpoint shard if the checkpoint is sharded. # We'll need to download and cache each checkpoint shard if the checkpoint is sharded.
if is_sharded: if is_sharded:
# rsolved_archive_file becomes a list of files that point to the different checkpoint shards in this case. # resolved_archive_file becomes a list of files that point to the different checkpoint shards in this case.
resolved_archive_file, sharded_metadata = get_checkpoint_shard_files( resolved_archive_file, sharded_metadata = get_checkpoint_shard_files(
pretrained_model_name_or_path, pretrained_model_name_or_path,
resolved_archive_file, resolved_archive_file,
......
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