Update generate.py

generate.py

@@ -7,7 +7,7 @@ from huggingface_hub import snapshot_download
 # Arguments
 parser = argparse.ArgumentParser()
 parser.add_argument("--task", type=str, default="t2v-1.3B")
-parser.add_argument("--size", type=str, default="832*480")
+parser.add_argument("--size", type=str, default="200*200")
 parser.add_argument("--frame_num", type=int, default=60)
 parser.add_argument("--sample_steps", type=int, default=20)
 parser.add_argument("--ckpt_dir", type=str, default="./Wan2.1-T2V-1.3B")
@@ -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 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}\""
+command = f"python generate.py --task t2v-1.3B --size 200*200 --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)