alex commited on
Commit
6d1382e
·
1 Parent(s): 3d2afe7

reduced sample steps

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. wan/configs/wan_animate_14B.py +1 -1
app.py CHANGED
@@ -220,7 +220,7 @@ def calculate_time_required(input_video, rc_bool):
220
  pose2d_tracking_duration_s = 50
221
  iteration_per_step_s = 12
222
 
223
- time_required = pose2d_tracking_duration_s + iteration_per_step_s * 20 * chunks
224
  print(f'for frames_count:{frames_count} doing {chunks} chunks the time_required is {time_required}')
225
  return time_required
226
 
 
220
  pose2d_tracking_duration_s = 50
221
  iteration_per_step_s = 12
222
 
223
+ time_required = pose2d_tracking_duration_s + iteration_per_step_s * 10 * chunks
224
  print(f'for frames_count:{frames_count} doing {chunks} chunks the time_required is {time_required}')
225
  return time_required
226
 
wan/configs/wan_animate_14B.py CHANGED
@@ -33,7 +33,7 @@ animate_14B.motion_encoder_dim = 512
33
 
34
  # inference
35
  animate_14B.sample_shift = 5.0
36
- animate_14B.sample_steps = 20
37
  animate_14B.sample_guide_scale = 1.0
38
  animate_14B.frame_num = 77
39
  animate_14B.sample_fps = 30
 
33
 
34
  # inference
35
  animate_14B.sample_shift = 5.0
36
+ animate_14B.sample_steps = 10
37
  animate_14B.sample_guide_scale = 1.0
38
  animate_14B.frame_num = 77
39
  animate_14B.sample_fps = 30