Skip to content

Commit 0671ba1

Browse files
committed
update
1 parent 6e2a93d commit 0671ba1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/loaders/single_file_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -397,6 +397,7 @@ def load_single_file_checkpoint(
397397

398398
else:
399399
repo_id, weights_name = _extract_repo_id_and_weights_name(pretrained_model_link_or_path)
400+
user_agent = {"file_type": "single_file", "framework": "pytorch"}
400401
pretrained_model_link_or_path = _get_model_file(
401402
repo_id,
402403
weights_name=weights_name,
@@ -406,6 +407,7 @@ def load_single_file_checkpoint(
406407
local_files_only=local_files_only,
407408
token=token,
408409
revision=revision,
410+
user_agent=user_agent,
409411
)
410412

411413
checkpoint = load_state_dict(pretrained_model_link_or_path, disable_mmap=disable_mmap)

0 commit comments

Comments
 (0)