Spaces:
Runtime error
Runtime error
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 />
|