up

yiyixuxu · yiyixuxu · commit e452ff1e5cb5 · 2025-06-22T13:04:21.000+02:00
diff --git a/src/diffusers/loaders/lora_base.py b/src/diffusers/loaders/lora_base.py
@@ -442,7 +442,8 @@ def _func_optionally_disable_offloading(_pipeline):
                 logger.info(
                     "Accelerate hooks detected. Since you have called `load_lora_weights()`, the previous hooks will be first removed. Then the LoRA parameters will be loaded and the hooks will be applied again."
                 )
-                remove_hook_from_module(component, recurse=is_sequential_cpu_offload)
+                if is_sequential_cpu_offload or is_model_cpu_offload:
+                    remove_hook_from_module(component, recurse=is_sequential_cpu_offload)
 
     return (is_model_cpu_offload, is_sequential_cpu_offload)
 
diff --git a/src/diffusers/loaders/textual_inversion.py b/src/diffusers/loaders/textual_inversion.py
@@ -427,7 +427,8 @@ def load_textual_inversion(
                         logger.info(
                             "Accelerate hooks detected. Since you have called `load_textual_inversion()`, the previous hooks will be first removed. Then the textual inversion parameters will be loaded and the hooks will be applied again."
                         )
-                        remove_hook_from_module(component, recurse=is_sequential_cpu_offload)
+                        if is_sequential_cpu_offload or is_model_cpu_offload:
+                            remove_hook_from_module(component, recurse=is_sequential_cpu_offload)
 
         # 7.2 save expected device and dtype
         device = text_encoder.device

Original file line number	Diff line number	Diff line change
`@@ -442,7 +442,8 @@ def _func_optionally_disable_offloading(_pipeline):`
`442`	`442`	`logger.info(`
`443`	`443`	"Accelerate hooks detected. Since you have called `load_lora_weights()`, the previous hooks will be first removed. Then the LoRA parameters will be loaded and the hooks will be applied again."
`444`	`444`	`)`
`445`		`- remove_hook_from_module(component, recurse=is_sequential_cpu_offload)`
	`445`	`+ if is_sequential_cpu_offload or is_model_cpu_offload:`
	`446`	`+ remove_hook_from_module(component, recurse=is_sequential_cpu_offload)`
`446`	`447`
`447`	`448`	`return (is_model_cpu_offload, is_sequential_cpu_offload)`
`448`	`449`
Original file line number	Diff line number	Diff line change
`@@ -427,7 +427,8 @@ def load_textual_inversion(`
`427`	`427`	`logger.info(`
`428`	`428`	"Accelerate hooks detected. Since you have called `load_textual_inversion()`, the previous hooks will be first removed. Then the textual inversion parameters will be loaded and the hooks will be applied again."
`429`	`429`	`)`
`430`		`- remove_hook_from_module(component, recurse=is_sequential_cpu_offload)`
	`430`	`+ if is_sequential_cpu_offload or is_model_cpu_offload:`
	`431`	`+ remove_hook_from_module(component, recurse=is_sequential_cpu_offload)`
`431`	`432`
`432`	`433`	`# 7.2 save expected device and dtype`
`433`	`434`	`device = text_encoder.device`