FireFlow

Running on Zero

wjs0725 commited on 30 days ago

Commit

6ffa574

•

1 Parent(s): 7fd2322

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ class FluxEditor:
         self.model.cuda()
     @torch.inference_mode()
-    @spaces.GPU(duration=180)
     def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
         torch.cuda.empty_cache()
         seed = None

         self.model.cuda()
     @torch.inference_mode()
+    @spaces.GPU(duration=140)
     def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
         torch.cuda.empty_cache()
         seed = None