Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
49c7172
1
Parent(s):
3f45c7e
Update app.py
Browse files
app.py
CHANGED
|
@@ -135,8 +135,6 @@ def generate(
|
|
| 135 |
pipe.load_lora_weights(lora2, adapter_name="2")
|
| 136 |
pipe.set_adapters(["1", "2"], adapter_weights=[lora_scale, lora_scale2])
|
| 137 |
|
| 138 |
-
reset_device_map()
|
| 139 |
-
pipe.enable_model_cpu_offload()
|
| 140 |
generator = torch.Generator().manual_seed(seed)
|
| 141 |
|
| 142 |
if not use_negative_prompt:
|
|
|
|
| 135 |
pipe.load_lora_weights(lora2, adapter_name="2")
|
| 136 |
pipe.set_adapters(["1", "2"], adapter_weights=[lora_scale, lora_scale2])
|
| 137 |
|
|
|
|
|
|
|
| 138 |
generator = torch.Generator().manual_seed(seed)
|
| 139 |
|
| 140 |
if not use_negative_prompt:
|