QZFantasies commited on
Commit
ae211e5
·
1 Parent(s): a8dfdef
Files changed (1) hide show
  1. LHM/utils/ffmpeg_utils.py +1 -1
LHM/utils/ffmpeg_utils.py CHANGED
@@ -54,7 +54,7 @@ def images_to_video(images, output_path, fps, gradio_codec: bool, verbose=False,
54
  # frames.append(cv2.resize(frame, (width // 2, height // 2)))
55
 
56
  # limit the frames directly @NOTE huggingface only!
57
- frames = frames[:300]
58
 
59
  frames = np.stack(frames)
60
  iio.imwrite(output_path,frames,fps=fps,codec="libx264",pixelformat="yuv420p",bitrate=bitrate,macro_block_size=16)
 
54
  # frames.append(cv2.resize(frame, (width // 2, height // 2)))
55
 
56
  # limit the frames directly @NOTE huggingface only!
57
+ frames = frames[:300]
58
 
59
  frames = np.stack(frames)
60
  iio.imwrite(output_path,frames,fps=fps,codec="libx264",pixelformat="yuv420p",bitrate=bitrate,macro_block_size=16)