calculate-model-flops / requirements.txt
MrYxJ
create
62d807b
raw history blame
No virus
172 Bytes
calflops @ git+https://github.com/MrYxJ/calculate-flops.pytorch
accelerate @ git+https://github.com/huggingface/accelerate
transformers
timm
huggingface_hub
tabulate
pandas