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