Update generate.py

generate.py

@@ -43,7 +43,7 @@ if torch.cuda.is_available():
 
 # Command to run the model generation
 offload_model_value = "True" if args.offload_model else "False"
-command = f"python generate.py --task t2v-1.3B --size 200*200 --frame_num 1 --ckpt_dir ./Wan2.1-T2V-1.3B --offload_model {offload_model_value} --t5_cpu --sample_shift 8 --sample_guide_scale 6 --prompt \"{args.prompt}\""
+command = f"python generate.py --task t2v-1.3B --size 480832 --frame_num 1 --ckpt_dir ./Wan2.1-T2V-1.3B --offload_model True --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)