Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -89,7 +89,7 @@ class FluxEditor:
|
|
| 89 |
self.ae.encoder.to(self.device)
|
| 90 |
|
| 91 |
@torch.inference_mode()
|
| 92 |
-
@spaces.GPU(duration=
|
| 93 |
def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
|
| 94 |
torch.cuda.empty_cache()
|
| 95 |
seed = None
|
|
|
|
| 89 |
self.ae.encoder.to(self.device)
|
| 90 |
|
| 91 |
@torch.inference_mode()
|
| 92 |
+
@spaces.GPU(duration=150)
|
| 93 |
def edit(self, init_image, source_prompt, target_prompt, num_steps, inject_step, guidance, seed):
|
| 94 |
torch.cuda.empty_cache()
|
| 95 |
seed = None
|