rynmurdock commited on
Commit
59f9d35
1 Parent(s): bfd5a13

mysterious spaces-only error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ pipe.to(device=DEVICE)
56
  # TODO put back
57
  @spaces.GPU
58
  def compile_em():
59
- pipe.unet = torch.compile(pipe.unet, mode='reduce-overhead')
60
  pipe.vae = torch.compile(pipe.vae, mode='reduce-overhead')
61
  autoencoder.model.forward = torch.compile(autoencoder.model.forward, backend='inductor', dynamic=True)
62
 
 
56
  # TODO put back
57
  @spaces.GPU
58
  def compile_em():
59
+ pipe.unet = torch.compile(pipe.unet)
60
  pipe.vae = torch.compile(pipe.vae, mode='reduce-overhead')
61
  autoencoder.model.forward = torch.compile(autoencoder.model.forward, backend='inductor', dynamic=True)
62