torch>=1.9.0
torchaudio~=0.9.0
torchvision~=0.10.0
pytorch-lightning~=1.3.6
lightning-bolts~=0.3.3
numpy~=1.20.1
fn~=0.4.3
einops~=0.3.0
av~=8.0.3
matplotlib~=3.3.4
clean-fid~=0.1.12
