muellerzr HF staff commited on
Commit
5a98202
·
1 Parent(s): 2c142c8

Use accelerate

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
- model-sizer @ git+https://github.com/muellerzr/model-sizer@main
2
  transformers
3
  timm
4
  huggingface_hub
 
1
+ accelerate @ git+https://github.com/huggingface/accelerate@model-size-estimator
2
  transformers
3
  timm
4
  huggingface_hub