Spaces:
Paused
Paused
Update inference.py
Browse files- inference.py +2 -1
inference.py
CHANGED
@@ -57,6 +57,7 @@ def get_args():
|
|
57 |
parser.add_argument("--smoother_steps", nargs='+', default=[19, 20], type=int, help="Timesteps at which using interleaved-frame smoother")
|
58 |
parser.add_argument("--is_long_video", action='store_true', help="Whether to use hierarchical sampler to produce long video")
|
59 |
parser.add_argument("--seed", type=int, default=42, help="Random seed of generator")
|
|
|
60 |
|
61 |
args = parser.parse_args()
|
62 |
return args
|
@@ -132,4 +133,4 @@ if __name__ == "__main__":
|
|
132 |
generator=generator, guidance_scale=12.5, negative_prompt=NEG_PROMPT,
|
133 |
width=args.width, height=args.height
|
134 |
).videos
|
135 |
-
save_videos_grid(sample, f"{args.output_path}/{args.
|
|
|
57 |
parser.add_argument("--smoother_steps", nargs='+', default=[19, 20], type=int, help="Timesteps at which using interleaved-frame smoother")
|
58 |
parser.add_argument("--is_long_video", action='store_true', help="Whether to use hierarchical sampler to produce long video")
|
59 |
parser.add_argument("--seed", type=int, default=42, help="Random seed of generator")
|
60 |
+
parser.add_argument("--temp_chunk_path", type=str, required=True, help="Path to temporary chunks")
|
61 |
|
62 |
args = parser.parse_args()
|
63 |
return args
|
|
|
133 |
generator=generator, guidance_scale=12.5, negative_prompt=NEG_PROMPT,
|
134 |
width=args.width, height=args.height
|
135 |
).videos
|
136 |
+
save_videos_grid(sample, f"{args.output_path}/{args.temp_chunk_path}.mp4")
|