Commit
•
ee58b95
1
Parent(s):
b2d7205
Update app.py
Browse files
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 |
|