Upload app.py
Browse files
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=
|
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=
|
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 |
)
|