calculate-model-flops / requirements.txt
MrYxJ
create
28cab40
raw
history blame
139 Bytes
calflops
bitsandbytes==0.41.1
accelerate @ git+https://github.com/huggingface/accelerate
transformers
timm
huggingface_hub
tabulate
pandas