Muhammad Taqi Raza
commited on
Commit
·
c6789b3
1
Parent(s):
cc317e3
fps=24
Browse files- 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=
|
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 |
|