fffiloni commited on
Commit
f7e7660
·
verified ·
1 Parent(s): 7255ed6

Update app_df.py

Browse files
Files changed (1) hide show
  1. app_df.py +7 -7
app_df.py CHANGED
@@ -21,16 +21,16 @@ if is_shared_ui:
21
  def generate_diffusion_forced_video(
22
  prompt,
23
  image=None,
24
- target_length="4",
25
  model_id="Skywork/SkyReels-V2-DF-1.3B-540P",
26
  resolution="540P",
27
- num_frames=97,
28
  ar_step=0,
29
  causal_attention=False,
30
  causal_block_size=1,
31
  base_num_frames=97,
32
- overlap_history=None,
33
- addnoise_condition=0,
34
  guidance_scale=6.0,
35
  shift=8.0,
36
  inference_steps=30,
@@ -39,9 +39,9 @@ def generate_diffusion_forced_video(
39
  fps=24,
40
  seed=None,
41
  prompt_enhancer=False,
42
- teacache=False,
43
  teacache_thresh=0.2,
44
- use_ret_steps=False,
45
  ):
46
  model_id = download_model(model_id)
47
 
@@ -246,7 +246,7 @@ with gr.Blocks() as demo:
246
  overlap_history = 17
247
  addnoise_condition = 20
248
  use_teacache = True
249
- teacache_thresh = 0.3
250
  use_ret_steps = True
251
  elif target_l == "15":
252
  n_frames = 377
 
21
  def generate_diffusion_forced_video(
22
  prompt,
23
  image=None,
24
+ target_length="10",
25
  model_id="Skywork/SkyReels-V2-DF-1.3B-540P",
26
  resolution="540P",
27
+ num_frames=257,
28
  ar_step=0,
29
  causal_attention=False,
30
  causal_block_size=1,
31
  base_num_frames=97,
32
+ overlap_history=17,
33
+ addnoise_condition=20,
34
  guidance_scale=6.0,
35
  shift=8.0,
36
  inference_steps=30,
 
39
  fps=24,
40
  seed=None,
41
  prompt_enhancer=False,
42
+ teacache=True,
43
  teacache_thresh=0.2,
44
+ use_ret_steps=True,
45
  ):
46
  model_id = download_model(model_id)
47
 
 
246
  overlap_history = 17
247
  addnoise_condition = 20
248
  use_teacache = True
249
+ teacache_thresh = 0.2
250
  use_ret_steps = True
251
  elif target_l == "15":
252
  n_frames = 377