Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -48,12 +48,12 @@ if torch.cuda.is_available():
|
|
48 |
"madebyollin/taesd3", torch_dtype=torch.float16
|
49 |
)
|
50 |
|
51 |
-
pipe.vae.decoder.layers = torch.compile(
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
)
|
57 |
pipe.vae.config.shift_factor = 0.0
|
58 |
|
59 |
pipe = pipe.to(device)
|
|
|
48 |
"madebyollin/taesd3", torch_dtype=torch.float16
|
49 |
)
|
50 |
|
51 |
+
# pipe.vae.decoder.layers = torch.compile(
|
52 |
+
# pipe.vae.decoder.layers,
|
53 |
+
# fullgraph=True,
|
54 |
+
# dynamic=False,
|
55 |
+
# mode="max-autotune-no-cudagraphs",
|
56 |
+
# )
|
57 |
pipe.vae.config.shift_factor = 0.0
|
58 |
|
59 |
pipe = pipe.to(device)
|