jiuface commited on
Commit
98e10c3
1 Parent(s): 12b22a2

Update lora_loading_patch.py

Browse files
Files changed (1) hide show
  1. 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 />