Update app.py
Browse files
app.py
CHANGED
|
@@ -39,7 +39,9 @@ def feifeimodload():
|
|
| 39 |
adapter_name=["feifei"],
|
| 40 |
lora_scale=1.0,
|
| 41 |
)
|
| 42 |
-
|
|
|
|
|
|
|
| 43 |
pipe.unload_lora_weights()
|
| 44 |
torch.cuda.empty_cache()
|
| 45 |
return pipe
|
|
|
|
| 39 |
adapter_name=["feifei"],
|
| 40 |
lora_scale=1.0,
|
| 41 |
)
|
| 42 |
+
# Enable VAE big pic
|
| 43 |
+
pipe.vae.enable_slicing()
|
| 44 |
+
pipe.vae.enable_tiling()
|
| 45 |
pipe.unload_lora_weights()
|
| 46 |
torch.cuda.empty_cache()
|
| 47 |
return pipe
|