requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu113

torch==1.12.1+cu113
torchvision==0.13.1+cu113

pyyaml
easydict

tqdm==4.60.0
transformers==4.19.2
torchmetrics==0.7.3

einops==0.3.0
omegaconf==2.1.1
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers