John6666 commited on
Commit
84880d5
1 Parent(s): 0799297

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -550,7 +550,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
550
  pipe_i2i.load_lora_weights(
551
  lora_path,
552
  weight_name=weight_name if weight_name else None,
553
- low_cpu_mem_usage=True,
554
  adapter_name=lora_name,
555
  token=HF_TOKEN
556
  )
@@ -558,7 +558,7 @@ def run_lora(prompt, image_input, image_strength, cfg_scale, steps, selected_ind
558
  pipe.load_lora_weights(
559
  lora_path,
560
  weight_name=weight_name if weight_name else None,
561
- low_cpu_mem_usage=True,
562
  adapter_name=lora_name,
563
  token=HF_TOKEN
564
  )
 
550
  pipe_i2i.load_lora_weights(
551
  lora_path,
552
  weight_name=weight_name if weight_name else None,
553
+ low_cpu_mem_usage=False,
554
  adapter_name=lora_name,
555
  token=HF_TOKEN
556
  )
 
558
  pipe.load_lora_weights(
559
  lora_path,
560
  weight_name=weight_name if weight_name else None,
561
+ low_cpu_mem_usage=False,
562
  adapter_name=lora_name,
563
  token=HF_TOKEN
564
  )