Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -39,7 +39,7 @@ pipe.fuse_lora(adapter_name=["better"], lora_scale=1.0)
|
|
| 39 |
pipe.unload_lora_weights()
|
| 40 |
|
| 41 |
# Memory optimizations
|
| 42 |
-
pipe.
|
| 43 |
pipe.enable_xformers_memory_efficient_attention() # Flash Attention
|
| 44 |
|
| 45 |
# CUDA Graph setup
|
|
|
|
| 39 |
pipe.unload_lora_weights()
|
| 40 |
|
| 41 |
# Memory optimizations
|
| 42 |
+
pipe.transformer.to(memory_format=torch.channels_last) # Channels last
|
| 43 |
pipe.enable_xformers_memory_efficient_attention() # Flash Attention
|
| 44 |
|
| 45 |
# CUDA Graph setup
|