fffiloni commited on
Commit
317c75b
·
verified ·
1 Parent(s): e381ffb

disable causal attention for the longest video lengths

Browse files
Files changed (1) hide show
  1. app_df.py +6 -6
app_df.py CHANGED
@@ -262,9 +262,9 @@ with gr.Blocks() as demo:
262
  use_teacache = True
263
  teacache_thresh = 0.3
264
  use_ret_steps = True
265
- causal_attention = True
266
- ar_step = 5
267
- causal_block_size = 5
268
  elif target_l == "60":
269
  n_frames = 1457
270
  overlap_history = 17
@@ -272,9 +272,9 @@ with gr.Blocks() as demo:
272
  use_teacache = True
273
  teacache_thresh = 0.3
274
  use_ret_steps = True
275
- causal_attention = True
276
- ar_step = 5
277
- causal_block_size = 5
278
 
279
  return n_frames, overlap_history, addnoise_condition, ar_step, causal_attention, causal_block_size, use_teacache, teacache_thresh, use_ret_steps
280
 
 
262
  use_teacache = True
263
  teacache_thresh = 0.3
264
  use_ret_steps = True
265
+ causal_attention = False
266
+ ar_step = 0
267
+ causal_block_size = 1
268
  elif target_l == "60":
269
  n_frames = 1457
270
  overlap_history = 17
 
272
  use_teacache = True
273
  teacache_thresh = 0.3
274
  use_ret_steps = True
275
+ causal_attention = False
276
+ ar_step = 0
277
+ causal_block_size = 0
278
 
279
  return n_frames, overlap_history, addnoise_condition, ar_step, causal_attention, causal_block_size, use_teacache, teacache_thresh, use_ret_steps
280