Spaces:
Running
on
Zero
Running
on
Zero
shuanholmes
commited on
Commit
·
8979df6
1
Parent(s):
29a2fa1
[FireFlow] Cache Demo Examples
Browse files
app.py
CHANGED
|
@@ -288,7 +288,11 @@ def create_demo(model_name: str, device: str = "cuda:0" if torch.cuda.is_availab
|
|
| 288 |
num_steps,
|
| 289 |
inject_step,
|
| 290 |
guidance
|
| 291 |
-
]
|
|
|
|
|
|
|
|
|
|
|
|
|
| 292 |
)
|
| 293 |
|
| 294 |
return demo
|
|
|
|
| 288 |
num_steps,
|
| 289 |
inject_step,
|
| 290 |
guidance
|
| 291 |
+
],
|
| 292 |
+
outputs=[output_image],
|
| 293 |
+
fn=edit,
|
| 294 |
+
cache_mode='eager',
|
| 295 |
+
cache_examples=True # Enable caching
|
| 296 |
)
|
| 297 |
|
| 298 |
return demo
|