--extra-index-url https://download.pytorch.org/whl/cu118 addict==2.4.0 albumentations==1.3.0 einops==0.6.0 gradio==3.21 huggingface-hub==0.13.2 imageio==2.26.0 imageio-ffmpeg==0.4.8 kornia omegaconf==2.3.0 open-clip-torch opencv-contrib-python==4.7.0.72 opencv-python-headless==4.7.0.72 prettytable==3.6.0 pytorch-lightning safetensors==0.3.0 timm https://download.pytorch.org/whl/cu118/torch-2.0.0%2Bcu118-cp39-cp39-linux_x86_64.whl https://download.pytorch.org/whl/cu118/torchvision-0.15.1%2Bcu118-cp39-cp39-linux_x86_64.whl transformers==4.27.1 datasets==2.9.0 accelerate evaluate==0.4.0 yapf==0.32.0 Triton==2.0.0 #diffusers git+https://github.com/huggingface/diffusers.git scipy ftfy psutil