Spaces:
Running
on
A10G
Running
on
A10G
Commit
•
99cf08b
1
Parent(s):
5c43323
compiling once more lol
Browse files
app.py
CHANGED
|
@@ -58,8 +58,6 @@ pipe.to(device=DEVICE)
|
|
| 58 |
# TODO put back
|
| 59 |
@spaces.GPU
|
| 60 |
def compile_em():
|
| 61 |
-
# TODO Compile
|
| 62 |
-
return None
|
| 63 |
pipe.unet = torch.compile(pipe.unet)
|
| 64 |
pipe.vae = torch.compile(pipe.vae, mode='reduce-overhead')
|
| 65 |
autoencoder.model.forward = torch.compile(autoencoder.model.forward, backend='inductor', dynamic=True)
|
|
|
|
| 58 |
# TODO put back
|
| 59 |
@spaces.GPU
|
| 60 |
def compile_em():
|
|
|
|
|
|
|
| 61 |
pipe.unet = torch.compile(pipe.unet)
|
| 62 |
pipe.vae = torch.compile(pipe.vae, mode='reduce-overhead')
|
| 63 |
autoencoder.model.forward = torch.compile(autoencoder.model.forward, backend='inductor', dynamic=True)
|