torch==2.3.1
torchvision==0.18.1
transformers==4.42.3
opencv-python-headless<4.10
peft<0.14.0
timm==1.0.9
einops==0.8.0
flash_attn
sentencepiece==0.2.0
mmengine<1