requirements.txt

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

torch==2.0.0+cu117
torchvision==0.15.1

pyyaml==5.4.1
easydict==1.9
protobuf==3.20.3
fsspec==2022.7.1

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

safetensors==0.3.1

opencv-python
scikit-image