requirements.txt

--extra-index-url https://download.pytorch.org/whl/cu117
torch==2.0.0+cu117
torchvision==0.15

numpy==1.26.4
gradio==5.15.0

transformers==4.48.2
torchmetrics==1.6.1

einops==0.3.0
omegaconf==2.1.1

pyyaml
easydict
tqdm