radames commited on
Commit
8ed1bef
1 Parent(s): 72bd0bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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
- 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)
 
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)