calculate-model-flops / requirements.txt
MrYXJ's picture
Update requirements.txt
3c2516d verified
raw
history blame
166 Bytes
accelerate @ git+https://github.com/huggingface/accelerate
bitsandbytes==0.41.1
sentencepiece
calflops>=0.3.2
einops
huggingface_hub
pandas
transformers
timm
tabulate