torch~=1.8.1
torchaudio~=0.8.1
torchvision~=0.9.1
pytorch-lightning~=1.3.2
lightning-bolts~=0.3.3
numpy~=1.20.3
fn~=0.4.3
einops~=0.3.0
av~=8.0.3
matplotlib~=3.3.4
clean-fid~=0.1.12
