wjs0725 commited on
Commit
566e04b
·
verified ·
1 Parent(s): b4b0e38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ class FluxEditor:
70
  self.model.eval()
71
 
72
  @torch.inference_mode()
73
- @spaces.GPU(duration=30)
74
  def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
75
  torch.cuda.empty_cache()
76
  seed = None
 
70
  self.model.eval()
71
 
72
  @torch.inference_mode()
73
+ @spaces.GPU(duration=180)
74
  def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
75
  torch.cuda.empty_cache()
76
  seed = None