wjs0725 commited on
Commit
323d2cd
1 Parent(s): 3d298d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ class FluxEditor:
89
  self.ae.encoder.to(self.device)
90
 
91
  @torch.inference_mode()
92
- @spaces.GPU(duration=60)
93
  def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
94
  torch.cuda.empty_cache()
95
  seed = None
 
89
  self.ae.encoder.to(self.device)
90
 
91
  @torch.inference_mode()
92
+ @spaces.GPU(duration=150)
93
  def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
94
  torch.cuda.empty_cache()
95
  seed = None