multimodalart HF staff commited on
Commit
ee58b95
1 Parent(s): b2d7205

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ pipe = AuraFlowPipeline.from_pretrained(
19
  torch_dtype=torch.float16
20
  ).to("cuda")
21
 
22
- pipe.transformer.to(memory_format=torch.channels_last)
23
- pipe.transformer = torch.compile(pipe.transformer, mode="reduce-overhead", fullgraph=True)
24
  #pipe.transformer.to(memory_format=torch.channels_last)
25
  #pipe.vae.to(memory_format=torch.channels_last)
26
 
 
19
  torch_dtype=torch.float16
20
  ).to("cuda")
21
 
22
+ #pipe.transformer.to(memory_format=torch.channels_last)
23
+ #pipe.transformer = torch.compile(pipe.transformer, mode="reduce-overhead", fullgraph=True)
24
  #pipe.transformer.to(memory_format=torch.channels_last)
25
  #pipe.vae.to(memory_format=torch.channels_last)
26