av.codec.codec.UnknownCodecError: libx264

[Greenhand-1]

File "/data/SEINE/sample_scripts/with_mask_sample.py", line 258, in
main(omega_conf)
File "/data/SEINE/sample_scripts/with_mask_sample.py", line 247, in main
torchvision.io.write_video(save_video_path, video_, fps=8)
File "/root/anaconda3/envs/seine/lib/python3.9/site-packages/torchvision/io/video.py", line 91, in write_video
stream = container.add_stream(video_codec, rate=fps)
File "av/container/output.pyx", line 67, in av.container.output.OutputContainer.add_stream
File "av/codec/codec.pyx", line 185, in av.codec.codec.Codec.cinit
File "av/codec/codec.pyx", line 194, in av.codec.codec.Codec._init
av.codec.codec.UnknownCodecError: libx264