Skip to content

Commit 7ee9465

Browse files
committed
modify _optionally_disable_offloading()
1 parent 7bfc1ee commit 7ee9465

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/loaders/lora.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -355,7 +355,7 @@ def _optionally_disable_offloading(cls, _pipeline):
355355
if not is_model_cpu_offload:
356356
is_model_cpu_offload = isinstance(component._hf_hook, CpuOffload)
357357
if not is_sequential_cpu_offload:
358-
is_sequential_cpu_offload = (
358+
is_sequential_cpu_offload = component.device.type == "cpu" and (
359359
isinstance(component._hf_hook, AlignDevicesHook)
360360
or hasattr(component._hf_hook, "hooks")
361361
and isinstance(component._hf_hook.hooks[0], AlignDevicesHook)

0 commit comments

Comments
 (0)