fix Number of Frames parameter

#2
by moskidon - opened
Files changed (1) hide show
  1. gradio_demo/runner.py +1 -1
gradio_demo/runner.py CHANGED
@@ -105,7 +105,7 @@ class Runner:
105
  config.input_data.sample_frame_rate = sample_frame_rate
106
 
107
  config.validation_data.prompts = [validation_prompt]
108
- config.validation_data.video_length = 8
109
  config.validation_data.width = resolution
110
  config.validation_data.height = resolution
111
  config.validation_data.num_inference_steps = 50
 
105
  config.input_data.sample_frame_rate = sample_frame_rate
106
 
107
  config.validation_data.prompts = [validation_prompt]
108
+ config.validation_data.video_length = n_sample_frames
109
  config.validation_data.width = resolution
110
  config.validation_data.height = resolution
111
  config.validation_data.num_inference_steps = 50