Spaces:
Running
on
A10G
Running
on
A10G
rynmurdock
commited on
Commit
•
59f9d35
1
Parent(s):
bfd5a13
mysterious spaces-only error
Browse files
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
|
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 |
|