-f https://download.pytorch.org/whl/torch_stable.html
torch==2.0.1+cu117
torchvision==0.15.2+cu117
lightning>=2.0
torchmetrics==1.2.0
numpy<2
pandas
h5py
yacs
omegaconf
einops
fvcore
scipy
scikit-learn
ninja
pillow
opencv-python
imageio
scikit-image
matplotlib
tensorboard
python-dotenv
diffusers==0.13.0
wandb>=0.12.10
taming-transformers==0.0.1
huggingface-hub==0.25.2