Skip to content

Commit 0f079b9

Browse files
authored
[Fix] Using sharded checkpoints with gated repositories (#9737)
fix
1 parent b0ffe92 commit 0f079b9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/utils/hub_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -457,7 +457,7 @@ def _get_checkpoint_shard_files(
457457
ignore_patterns = ["*.json", "*.md"]
458458
if not local_files_only:
459459
# `model_info` call must guarded with the above condition.
460-
model_files_info = model_info(pretrained_model_name_or_path, revision=revision)
460+
model_files_info = model_info(pretrained_model_name_or_path, revision=revision, token=token)
461461
for shard_file in original_shard_filenames:
462462
shard_file_present = any(shard_file in k.rfilename for k in model_files_info.siblings)
463463
if not shard_file_present:

0 commit comments

Comments
 (0)