We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6e2a93d commit 0671ba1Copy full SHA for 0671ba1
src/diffusers/loaders/single_file_utils.py
@@ -397,6 +397,7 @@ def load_single_file_checkpoint(
397
398
else:
399
repo_id, weights_name = _extract_repo_id_and_weights_name(pretrained_model_link_or_path)
400
+ user_agent = {"file_type": "single_file", "framework": "pytorch"}
401
pretrained_model_link_or_path = _get_model_file(
402
repo_id,
403
weights_name=weights_name,
@@ -406,6 +407,7 @@ def load_single_file_checkpoint(
406
407
local_files_only=local_files_only,
408
token=token,
409
revision=revision,
410
+ user_agent=user_agent,
411
)
412
413
checkpoint = load_state_dict(pretrained_model_link_or_path, disable_mmap=disable_mmap)
0 commit comments