Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -70,7 +70,7 @@ class FluxEditor:
|
|
70 |
self.model.eval()
|
71 |
|
72 |
@torch.inference_mode()
|
73 |
-
@spaces.GPU(duration=
|
74 |
def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
|
75 |
torch.cuda.empty_cache()
|
76 |
seed = None
|
|
|
70 |
self.model.eval()
|
71 |
|
72 |
@torch.inference_mode()
|
73 |
+
@spaces.GPU(duration=180)
|
74 |
def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
|
75 |
torch.cuda.empty_cache()
|
76 |
seed = None
|