fffiloni commited on
Commit
f7f5e52
·
1 Parent(s): c5773bb

Update inference.py

Browse files
Files changed (1) hide show
  1. 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.prompt}.mp4")
 
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")