Update generate.py

generate.py

@@ -103,7 +103,7 @@ def generate(args):
         video = wan_t2v.generate(
             args.prompt,
             size=SIZE_CONFIGS[args.size],
-            frame_num=args.frame_num,
+            frame_num=3,
             shift=args.sample_shift,
             sample_solver=args.sample_solver,
             sampling_steps=args.sample_steps,
@@ -157,7 +157,7 @@ def generate(args):
             args.prompt,
             img,
             max_area=MAX_AREA_CONFIGS[args.size],
-            frame_num=args.frame_num,
+            frame_num=3,
             shift=args.sample_shift,
             sample_solver=args.sample_solver,
             sampling_steps=args.sample_steps,