Spaces:
Running
on
A10G
Running
on
A10G
Commit
•
540e4d3
1
Parent(s):
e060f2d
compile
Browse files
app.py
CHANGED
|
@@ -55,7 +55,6 @@ pipe.to(device=DEVICE)
|
|
| 55 |
|
| 56 |
@spaces.GPU
|
| 57 |
def compile_em():
|
| 58 |
-
return None# TODO add back
|
| 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)
|
|
|
|
| 55 |
|
| 56 |
@spaces.GPU
|
| 57 |
def compile_em():
|
|
|
|
| 58 |
pipe.unet = torch.compile(pipe.unet, mode='reduce-overhead')
|
| 59 |
pipe.vae = torch.compile(pipe.vae, mode='reduce-overhead')
|
| 60 |
autoencoder.model.forward = torch.compile(autoencoder.model.forward, backend='inductor', dynamic=True)
|