Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -92,6 +92,7 @@ class FluxEditor:
|
|
92 |
|
93 |
@torch.inference_mode()
|
94 |
def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
|
|
|
95 |
torch.cuda.empty_cache()
|
96 |
seed = None
|
97 |
# if seed == -1:
|
|
|
92 |
|
93 |
@torch.inference_mode()
|
94 |
def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
|
95 |
+
|
96 |
torch.cuda.empty_cache()
|
97 |
seed = None
|
98 |
# if seed == -1:
|