Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -129,7 +129,7 @@ def prepare_data(device, input_image, x0, y0, x1, y1, polar, azimuth, text):
|
|
129 |
return batch
|
130 |
|
131 |
|
132 |
-
@spaces.GPU
|
133 |
@torch.no_grad()
|
134 |
def run_generation(sampler, model, device, input_image, x0, y0, x1, y1, polar, azimuth, text, seed):
|
135 |
seed_everything(seed)
|
|
|
129 |
return batch
|
130 |
|
131 |
|
132 |
+
@spaces.GPU(enable_queue=True)
|
133 |
@torch.no_grad()
|
134 |
def run_generation(sampler, model, device, input_image, x0, y0, x1, y1, polar, azimuth, text, seed):
|
135 |
seed_everything(seed)
|