AlekseyCalvin commited on
Commit
17b4f96
·
verified ·
1 Parent(s): fb103c9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -35,9 +35,9 @@ pipe = FluxWithCFGPipeline.from_pretrained(
35
  # pipe.set_adapters("fast")
36
  # pipe.fuse_lora(adapter_names=["fast"], lora_scale=1.0)
37
  pipe.to("cuda")
38
- pipe.transformer.to(memory_format=torch.channels_last)
39
- pipe.transformer = torch.compile(
40
- pipe.transformer, mode="max-autotune", fullgraph=True
41
  )
42
  torch.cuda.empty_cache()
43
 
 
35
  # pipe.set_adapters("fast")
36
  # pipe.fuse_lora(adapter_names=["fast"], lora_scale=1.0)
37
  pipe.to("cuda")
38
+ pipe.transformer.to(memory_format=torch.channels_last)
39
+ pipe.transformer = torch.compile(
40
+ pipe.transformer, mode="max-autotune", fullgraph=True
41
  )
42
  torch.cuda.empty_cache()
43