shuanholmes commited on
Commit
8979df6
1 Parent(s): 29a2fa1

[FireFlow] Cache Demo Examples

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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