Update modeling_utils.py (#9035 )

2025-12-07 13:04:15 +08:00 · 2024-11-17 06:46:56 +09:00
1 changed files with 1 additions and 1 deletions
--- a/src/diffusers/models/modeling_utils.py
+++ b/src/diffusers/models/modeling_utils.py
@@ -819,7 +819,7 @@ class ModelMixin(torch.nn.Module, PushToHubMixin):
                                offload_folder=offload_folder,
                                offload_state_dict=offload_state_dict,
                                dtype=torch_dtype,
-                                force_hook=force_hook,
+                                force_hooks=force_hook,
                                strict=True,
                            )
                            model._undo_temp_convert_self_to_deprecated_attention_blocks()