Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -114,7 +114,8 @@ def generate(
|
|
114 |
else:
|
115 |
adapter_name2 = cached_loras[lora_key2]
|
116 |
pipe.set_adapters([adapter_name1, adapter_name2], adapter_weights=[lora_scale, lora_scale2])
|
117 |
-
|
|
|
118 |
pipe.to(device)
|
119 |
generator = torch.Generator().manual_seed(seed)
|
120 |
|
|
|
114 |
else:
|
115 |
adapter_name2 = cached_loras[lora_key2]
|
116 |
pipe.set_adapters([adapter_name1, adapter_name2], adapter_weights=[lora_scale, lora_scale2])
|
117 |
+
|
118 |
+
enable_model_cpu_offload()
|
119 |
pipe.to(device)
|
120 |
generator = torch.Generator().manual_seed(seed)
|
121 |
|