Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
a2b9bf8
1
Parent(s):
b6711e8
Update app.py
Browse files
app.py
CHANGED
|
@@ -59,6 +59,7 @@ def generate(
|
|
| 59 |
|
| 60 |
if use_lora:
|
| 61 |
pipe.load_lora_weights(lora)
|
|
|
|
| 62 |
|
| 63 |
if ENABLE_CPU_OFFLOAD:
|
| 64 |
pipe.enable_model_cpu_offload()
|
|
|
|
| 59 |
|
| 60 |
if use_lora:
|
| 61 |
pipe.load_lora_weights(lora)
|
| 62 |
+
pipe.fuse_lora(lora_scale=0.7)
|
| 63 |
|
| 64 |
if ENABLE_CPU_OFFLOAD:
|
| 65 |
pipe.enable_model_cpu_offload()
|