Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -240,7 +240,7 @@ class main():
|
|
240 |
|
241 |
|
242 |
@torch.no_grad()
|
243 |
-
@spaces.GPU
|
244 |
def edit_inference(self, prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
|
245 |
device = self.device
|
246 |
self.unet.to(device)
|
|
|
240 |
|
241 |
|
242 |
@torch.no_grad()
|
243 |
+
@spaces.GPU(duration=120)
|
244 |
def edit_inference(self, prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
|
245 |
device = self.device
|
246 |
self.unet.to(device)
|