set max_sample_step to 60

app.py

@@ -34,7 +34,7 @@ def videocrafter_demo(result_dir='./tmp/'):
                         with gr.Row():
                             model_index = gr.Dropdown(label='Models', elem_id=f"model", choices=model_choices, value=model_choices[0], type="index",interactive=True)
                         with gr.Row():
-                            steps = gr.Slider(minimum=1, maximum=200, step=1, elem_id=f"steps", label="Sampling steps", value=50)
+                            steps = gr.Slider(minimum=1, maximum=60, step=1, elem_id=f"steps", label="Sampling steps", value=50)
                             eta = gr.Slider(minimum=0.0, maximum=1.0, step=0.1, label='ETA', value=1.0, elem_id="eta")
                         with gr.Row():
                             lora_scale = gr.Slider(minimum=0.0, maximum=2.0, step=0.1, label='Lora Scale', value=1.0, elem_id="lora_scale")
@@ -68,7 +68,7 @@ def videocrafter_demo(result_dir='./tmp/'):
                             vc_eta = gr.Slider(minimum=0.0, maximum=1.0, step=0.1, label='ETA', value=1.0, elem_id="vc_eta")
                             vc_cfg_scale = gr.Slider(minimum=1.0, maximum=30.0, step=0.5, label='CFG Scale', value=15.0, elem_id="vc_cfg_scale")
                         with gr.Row():
-                            vc_steps = gr.Slider(minimum=1, maximum=200, step=1, elem_id="vc_steps", label="Sampling steps", value=50)
+                            vc_steps = gr.Slider(minimum=1, maximum=60, step=1, elem_id="vc_steps", label="Sampling steps", value=50)
                             frame_stride = gr.Slider(minimum=0 , maximum=100, step=1, label='Frame Stride', value=0, elem_id="vc_frame_stride")
 
                         vc_end_btn = gr.Button("Send")

videocontrol_test.py

@@ -63,6 +63,8 @@ class VideoControl:
         self.spatial_transform = transforms_video.CenterCropVideo(self.resolution)
 
     def get_video(self, input_video, input_prompt, frame_stride=0, vc_steps=50, vc_cfg_scale=15.0, vc_eta=1.0):
+        if vc_steps > 60:
+            vc_steps = 60
         ## load video
         print("input video", input_video)
         info_str = ''

videocrafter_test.py

@@ -19,8 +19,8 @@ def save_results(videos, save_dir,
         npz_to_video_grid(videos[i:i+1,...], 
                             os.path.join(save_subdir, f"{save_name}_{i:03d}.mp4"), 
                             fps=save_fps)
-    print(f'Successfully saved videos in {save_subdir}')
     video_path_list = [os.path.join(save_subdir, f"{save_name}_{i:03d}.mp4") for i in range(videos.shape[0])]
+    print(f'Successfully saved videos in {video_path_list[0]}')
     return video_path_list
     
 
@@ -45,6 +45,8 @@ class Text2Video():
         self.origin_weight = None
 
     def get_prompt(self, input_text, steps=50, model_index=0, eta=1.0, cfg_scale=15.0, lora_scale=1.0):
+        if steps > 60:
+            steps = 60 
         if model_index > 0:
             input_text = input_text + ', ' + self.lora_trigger_word_list[model_index]
         inject_lora = model_index > 0