python-dotenv
datasets>=2.18.0
peft>=0.7.1
pip>=24.0
swanlab>=0.3.0
torch==2.10.0
torchvision
transformers>=5.1.0

[flash_attn]
flash-attn
