requirements.txt

accelerate @ git+https://github.com/huggingface/accelerate
bitsandbytes==0.41.1
sentencepiece
calflops>=0.3.2
einops
huggingface_hub
pandas
torch==2.2.0
transformers==4.38.2
timm
tiktoken
tabulate