Ryukijano commited on
Commit
eef4bc9
·
verified ·
1 Parent(s): cf48cc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.transformers.to(memory_format=torch.channels_last) # Channels last
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