FireFlow

Running on Zero

wjs0725 commited on 20 days ago

Commit

b95d1a4

•

1 Parent(s): 6ffa574

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=140)
     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=60)
     def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
         torch.cuda.empty_cache()
         seed = None