Muhammad Taqi Raza commited on
Commit
c6789b3
·
1 Parent(s): cc317e3
Files changed (1) hide show
  1. inference_script.py +2 -1
inference_script.py CHANGED
@@ -477,6 +477,7 @@ def process_video(
477
  image_rotary_emb=rotary_emb,
478
  return_dict=False,
479
  )[0]
 
480
 
481
  latent_generate = pipe.scheduler.get_velocity(
482
  predicted_noise, latent, timesteps
@@ -510,7 +511,7 @@ if __name__ == "__main__":
510
 
511
  parser.add_argument("--output_path", type=str, default="./results", help="The path save generated video")
512
 
513
- parser.add_argument("--fps", type=int, default=16, help="The frames per second for the generated video")
514
 
515
  parser.add_argument("--dtype", type=str, default="bfloat16", help="The data type for computation")
516
 
 
477
  image_rotary_emb=rotary_emb,
478
  return_dict=False,
479
  )[0]
480
+
481
 
482
  latent_generate = pipe.scheduler.get_velocity(
483
  predicted_noise, latent, timesteps
 
511
 
512
  parser.add_argument("--output_path", type=str, default="./results", help="The path save generated video")
513
 
514
+ parser.add_argument("--fps", type=int, default=24, help="The frames per second for the generated video")
515
 
516
  parser.add_argument("--dtype", type=str, default="bfloat16", help="The data type for computation")
517