--index-url https://download.pytorch.org/whl/cu118
torch
--index-url https://download.pytorch.org/whl/cu118 
torchvision
--index-url https://download.pytorch.org/whl/cu118 
torchaudio
transformers
scikit-image
matplotlib
diffusers
einops