FireFlow

Running on Zero

wjs0725 commited on 20 days ago

Commit

74a9853

•

1 Parent(s): b95d1a4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -79,7 +79,6 @@ class FluxEditor:
     @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
         # if seed == -1:
         #     seed = None

     @torch.inference_mode()
     @spaces.GPU(duration=60)
     def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
         seed = None
         # if seed == -1:
         #     seed = None