Update README.md

README.md

@@ -296,7 +296,7 @@ def split_video_into_frames(input_video_path, num_frames, temp_folder='temp_fram
 
     print(f"新的视频已保存到 resampled_video.mp4，包含 {num_frames} 个帧，并保持原始的帧率。")
 
-def generate_video_with_prompt(input_video_path, prompt, model_id, gif_output_path, seed=0, num_frames=16, keep_imgs=False, temp_folder='temp_frames'):
+def generate_video_with_prompt(input_video_path, prompt, model_id, gif_output_path, seed=0, num_frames=16, keep_imgs=False, temp_folder='temp_frames', num_inference_steps=50, guidance_scale=20, controlnet_conditioning_scale=1.0, width=512, height=768):
     """
     生成带有文本提示的视频。
 
@@ -308,6 +308,11 @@ def generate_video_with_prompt(input_video_path, prompt, model_id, gif_output_pa
     :param num_frames: 目标帧数
     :param keep_imgs: 是否保留临时图片
     :param temp_folder: 临时文件夹路径
+    :param num_inference_steps: 推理步数
+    :param guidance_scale: 引导比例
+    :param controlnet_conditioning_scale: ControlNet 条件比例
+    :param width: 输出宽度
+    :param height: 输出高度
     """
     split_video_into_frames(input_video_path, num_frames, temp_folder)
 
@@ -328,11 +333,11 @@ def generate_video_with_prompt(input_video_path, prompt, model_id, gif_output_pa
     output = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
-        num_inference_steps=50,
-        guidance_scale=20,
-        controlnet_conditioning_scale=1.0,
-        width=512,
-        height=768,
+        num_inference_steps=num_inference_steps,
+        guidance_scale=guidance_scale,
+        controlnet_conditioning_scale=controlnet_conditioning_scale,
+        width=width,
+        height=height,
         num_frames=num_frames,
         conditioning_frames=cn2,
         generator=generator
@@ -358,10 +363,16 @@ if __name__ == "__main__":
     parser.add_argument("--num_frames", type=int, default=16, help="目标帧数")
     parser.add_argument("--keep_imgs", action="store_true", help="是否保留临时图片")
     parser.add_argument("--temp_folder", default='temp_frames', help="临时文件夹路径")
+    parser.add_argument("--num_inference_steps", type=int, default=50, help="推理步数")
+    parser.add_argument("--guidance_scale", type=float, default=20.0, help="引导比例")
+    parser.add_argument("--controlnet_conditioning_scale", type=float, default=1.0, help="ControlNet 条件比例")
+    parser.add_argument("--width", type=int, default=512, help="输出宽度")
+    parser.add_argument("--height", type=int, default=768, help="输出高度")
 
     args = parser.parse_args()
 
-    generate_video_with_prompt(args.input_video, args.prompt, args.model_id, args.gif_output_path, args.seed, args.num_frames, args.keep_imgs, args.temp_folder)
+    generate_video_with_prompt(args.input_video, args.prompt, args.model_id, args.gif_output_path, args.seed, args.num_frames,
+      args.keep_imgs, args.temp_folder, args.num_inference_steps, args.guidance_scale, args.controlnet_conditioning_scale, args.width, args.height)
 ```
 
 ```bash