amildravid4292 commited on
Commit
5529fcd
·
verified ·
1 Parent(s): 99172cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)