Update generate.py

generate.py

@@ -41,11 +41,11 @@ if torch.cuda.is_available():
     torch.backends.cudnn.benchmark = False
     torch.backends.cudnn.deterministic = True
 
-# Run the model (Ensure that `generate.py` includes these new params in its model call)
-# Pass offload_model correctly as either True or False
+# Command to run the model generation
 offload_model_value = "True" if args.offload_model else "False"
-command = f"python generate.py --task {args.task} --size {args.size} --frame_num {args.frame_num} --sample_steps {args.sample_steps} --ckpt_dir {args.ckpt_dir} --offload_model {offload_model_value} --t5_cpu {args.t5_cpu} --sample_shift {args.sample_shift} --sample_guide_scale {args.sample_guide_scale} --prompt \"{args.prompt}\""
+command = f"python generate.py --task t2v-1.3B --size 832*480 --ckpt_dir ./Wan2.1-T2V-1.3B --offload_model {offload_model_value} --t5_cpu --sample_shift 8 --sample_guide_scale 6 --prompt \"{args.prompt}\""
 
+# Run the model
 process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
 stdout, stderr = process.communicate()