Spaces:
Running
on
Zero
Running
on
Zero
Update lora_loading_patch.py
Browse files- lora_loading_patch.py +3 -1
lora_loading_patch.py
CHANGED
@@ -1,3 +1,4 @@
|
|
|
|
1 |
from diffusers.utils import (
|
2 |
convert_unet_state_dict_to_peft,
|
3 |
get_peft_kwargs,
|
@@ -113,4 +114,5 @@ def load_lora_into_transformer(
|
|
113 |
if is_model_cpu_offload:
|
114 |
_pipeline.enable_model_cpu_offload()
|
115 |
elif is_sequential_cpu_offload:
|
116 |
-
_pipeline.enable_sequential_cpu_offload()
|
|
|
|
1 |
+
# ruff: noqa
|
2 |
from diffusers.utils import (
|
3 |
convert_unet_state_dict_to_peft,
|
4 |
get_peft_kwargs,
|
|
|
114 |
if is_model_cpu_offload:
|
115 |
_pipeline.enable_model_cpu_offload()
|
116 |
elif is_sequential_cpu_offload:
|
117 |
+
_pipeline.enable_sequential_cpu_offload()
|
118 |
+
# Unsafe code />
|