icecream == 2.1.3
transformers >= 4.37.2
flash-attn >= 2.3.6