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 |
|