--extra-index-url https://download.pytorch.org/whl/cu113
torch
torchvision
einops
ema-pytorch 
fsspec 
fvcore 
huggingface-hub 
matplotlib 
numpy==1.26.4 
opencv-python 
omegaconf 
pytorch-msssim 
scikit-image 
scikit-learn 
scipy 
tensorboard 
tensorboardx 
wandb 
timm