Mamba-Trainer / requirements.txt
Pratik Dwivedi
trainer commit (#1)
56d31bf
raw
history blame
103 Bytes
packaging
torch==2.2.1+cu121
transformers
causal-conv1d
mamba-ssm
accelerate
bitsandbytes
scipy==1.11.4