Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -109,6 +109,7 @@ class ModelWrapper:
|
|
| 109 |
|
| 110 |
|
| 111 |
@torch.no_grad()
|
|
|
|
| 112 |
def inference(self, prompt, seed, height, width, num_images, fast_vae_decode):
|
| 113 |
print("Running model inference...")
|
| 114 |
|
|
|
|
| 109 |
|
| 110 |
|
| 111 |
@torch.no_grad()
|
| 112 |
+
@spaces.GPU()
|
| 113 |
def inference(self, prompt, seed, height, width, num_images, fast_vae_decode):
|
| 114 |
print("Running model inference...")
|
| 115 |
|