numpy>=1.26.0
pytest
pytest-timeout>=2.2.0
torch>=2.0
torchvision
ipykernel
accelerate
diffusers
transformers
timm
datasets

[frozen_huggingface]
accelerate==1.1.0
diffusers==0.31.0
transformers==4.46.3
timm==0.9.16
datasets==3.1.0
