Spaces:
Running
on
A10G
Running
on
A10G
Commit
•
f5c8f7e
1
Parent(s):
b670aa6
__pycache__/safety_checker_improved.cpython-310.pyc
ADDED
|
Binary file (1.37 kB). View file
|
|
|
app.py
CHANGED
|
@@ -58,9 +58,9 @@ pipe.to(device=DEVICE)
|
|
| 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)
|
| 64 |
|
| 65 |
|
| 66 |
output_hidden_state = False
|
|
|
|
| 58 |
# TODO put back
|
| 59 |
@spaces.GPU
|
| 60 |
def compile_em():
|
| 61 |
+
pipe.unet = torch.compile(pipe.unet, mode='reduce-overhead')
|
| 62 |
pipe.vae = torch.compile(pipe.vae, mode='reduce-overhead')
|
| 63 |
+
autoencoder.model.forward = torch.compile(autoencoder.model.forward, backend='inductor', dynamic=True, mode='reduce-overhead')
|
| 64 |
|
| 65 |
|
| 66 |
output_hidden_state = False
|