# For GPU (CUDA 11.8):
# pip install torch==2.7.1 torchvision==0.22.1 --index-url https://download.pytorch.org/whl/cu118

# For CPU:
# torch==2.7.1
# torchvision==0.22.1

accelerate==1.8.1
datasets==4.0.0
diffusers==0.34.0
ipywidgets==8.1.7
kornia==0.8.1
matplotlib==3.10.5
numpy==2.2.6
notebook==7.4.4
omegaconf==2.3.0
Pillow==11.2.1
PyYAML==6.0.2
reportlab==4.4.3
scikit-learn==1.6.1
torch-fidelity==0.3.0
torchmetrics==1.8.1
tqdm==4.67.1
transformers==4.53.1
git+https://github.com/openai/CLIP.git