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.24.0
torchmetrics==0.7.3

einops==0.3.0
omegaconf==2.1.1
huggingface-hub==0.11.1
gradio==3.17.1