huggingface/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-05-28 00:39:35 +08:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
DN6
2026-05-23 00:55:18 +05:30
parent b4302317d6
commit 5eaaa3f0ea
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/diffusers/models/modeling_utils.py
View File

@@ -1635,10 +1635,6 @@ class ModelMixin(torch.nn.Module, ConfigMixin, PushToHubMixin):
# We only fix it for non sharded checkpoints as we don't need it yet for sharded one.
model._fix_state_dict_keys_on_load(state_dict)
# Convert checkpoint if needed (e.g., original format to diffusers format). For models that haven't
# registered weight-mapping metadata this is a no-op via the default handler.
state_dict = cls._weight_mapping.maybe_convert_state_dict(model, state_dict)
if is_sharded:
loaded_keys = sharded_metadata["all_checkpoint_keys"]
else: