requirements.txt

absl-py==0.13.0
aiohttp==3.7.4.post0
antlr4-python3-runtime==4.8
async-timeout==3.0.1
attrs==21.2.0
backcall==0.2.0
cachetools==4.2.2
certifi==2021.5.30
chardet==4.0.0
decorator==5.0.9
einops==0.3.0
fsspec==2021.6.1
ftfy==6.0.3
future==0.18.2
google-auth==1.32.0
google-auth-oauthlib==0.4.4
grpcio==1.38.1
idna==2.10
imageio==2.9.0
imageio-ffmpeg==0.4.4
ipython==7.25.0
ipython-genutils==0.2.0
jedi==0.18.0
kornia==0.5.4
Markdown==3.3.4
matplotlib-inline==0.1.2
multidict==5.1.0
numpy==1.21.0
oauthlib==3.1.1
omegaconf==2.1.0
packaging==20.9
parso==0.8.2
pexpect==4.8.0
pickleshare==0.7.5
Pillow==8.2.0
prompt-toolkit==3.0.19
protobuf==3.17.3
ptyprocess==0.7.0
pyasn1==0.4.8
pyasn1-modules==0.2.8
pyDeprecate==0.3.0
Pygments==2.9.0
pyparsing==2.4.7
pytorch-lightning==1.3.7.post0
PyYAML==6.0
regex==2021.4.4
requests==2.25.1
requests-oauthlib==1.3.0
rsa==4.7.2
six==1.16.0
tensorboard==2.4.1
tensorboard-plugin-wit==1.8.0
torch==1.9.0+cu111
torchaudio==0.9.0
torchmetrics==0.3.2
torchvision==0.10.0+cu111
torch-optimizer==0.1.0
tqdm==4.61.1
traitlets==5.0.5
typing-extensions==3.10.0.0
urllib3==1.26.6
wcwidth==0.2.5
Werkzeug==2.0.1
yarl==1.6.3
rudalle==0.0.1rc11