multimodalart HF Staff commited on
Commit
134049c
·
verified ·
1 Parent(s): bc88d24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -222,10 +222,17 @@ def rewrite_prompt_external(user_prompt, pil_image):
222
 
223
  class ArgsNamespace:
224
  def __init__(self):
225
- self.use_sageattn = False
226
  self.sage_blocks_range = "0-53"
 
 
227
  self.enable_torch_compile = False
228
- self.no_cache_block_id = ""
 
 
 
 
 
 
229
 
230
  initialize_infer_state(ArgsNamespace())
231
 
 
222
 
223
  class ArgsNamespace:
224
  def __init__(self):
 
225
  self.sage_blocks_range = "0-53"
226
+ self.no_cache_block_id = "0-0"
227
+ self.use_sageattn = False
228
  self.enable_torch_compile = False
229
+ self.enable_cache = False
230
+ self.cache_type = "deepcache"
231
+ self.cache_start_step = 11
232
+ self.cache_end_step = 45
233
+ self.total_steps = 50
234
+ self.cache_step_interval = 4
235
+
236
 
237
  initialize_infer_state(ArgsNamespace())
238