Spaces:
Runtime error
Runtime error
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)
|