model-memory-usage / requirements.txt
muellerzr's picture
muellerzr HF staff
Use accelerate
5a98202
raw
history blame
131 Bytes
accelerate @ git+https://github.com/huggingface/accelerate@model-size-estimator
transformers
timm
huggingface_hub
py-markdown-table