requirements.txt

# pytorch
# conda-forge
defaults
# cudatoolkit==11.3.1
# python==3.9.17
# pytorch==1.10.1
# torchvision==0.11.2
pip
accelerate==0.21.0
annotated-types==0.5.0
appdirs==1.4.4
attrs==23.1.0
autopep8==2.0.2
certifi==2023.7.22
charset-normalizer==3.2.0
cityscapesscripts==2.2.2
click==8.1.6
coloredlogs==15.0.1
contourpy==1.1.0
cycler==0.11.0
docker-pycreds==0.4.0
einops==0.6.1
einops-exts==0.0.4
ema-pytorch==0.2.3
filelock==3.12.2
fonttools==4.41.1
fsspec==2023.6.0
gitdb==4.0.10
gitpython==3.1.32
huggingface-hub==0.16.4
humanfriendly==10.0
idna==3.4
importlib-resources==6.0.0
jsonlines==3.1.0
kiwisolver==1.4.4
kornia==0.6.12
matplotlib==3.7.2
packaging==23.1
pathtools==0.1.2
protobuf==4.23.4
psutil==5.9.5
pycodestyle==2.11.0
pydantic==2.1.1
pydantic-core==2.4.0
pyparsing==3.0.9
pyquaternion==0.9.9
python-dateutil==2.8.2
pytorch-warmup==0.1.1
pyyaml==6.0.1
regex==2023.6.3
requests==2.31.0
safetensors==0.3.1
sentencepiece==0.1.99
sentry-sdk==1.28.1
setproctitle==1.3.2
smmap==5.0.0
tokenizers==0.13.3
tomli==2.0.1
tqdm==4.65.0
transformers==4.31.0
typing==3.7.4.3
urllib3==2.0.4
wandb==0.15.7
zipp==3.16.2