torch==2.2.2
numpy>=1.26.0,<=1.26.4
lightning>=2.4.0
wandb>=0.16.6
pyyaml>=6.0.1
torchvision>=0.17.2
datasets>=2.19.0
transformers>=4.40.1
python-dotenv>=1.0.1
requests>=2.32.3
uvicorn>=0.30.5
fastapi>=0.112.0
huggingface-hub>=0.23.4
scikit-learn>=1.5.1
strenum>=0.4.15