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

torch==2.0.1
torchvision
tensorboard
timm==0.5.4
opencv-python
setuptools==58.2.0
easydict
wget
scikit-image
fairscale
imgaug
peft
kornia
gradio==4.44.1
gradio_image_prompter
huggingface_hub