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
|