Introduce model memory estimator #1876
muellerzr
merged
34 commits into
main
from
model-size-estimator
Estimator
c371f033
Right err
f71d604c
Fixup tests
fe91f09a
trust remote code
7080016f
Print output for debugging purposes
e1fadd88
trust_remote_code
3b3e259e
Address some comments
e01d333a
change doc to req arg
6f4be35d
Properly check for _no_split_modules in transformer models
19050713
Note on transformer models
529c2ccf
sgugger
commented
on 2023-08-23
Check/handle pentabytes
36c89a18
Tests are passing locally again, better handle for no_split
373c9f2a
Merge branch 'model-size-estimator' of https://www.github.com/hugging…
93603644
Adjust setup?
7bb7a4c2
Let's see if the cleaner version works
fa796e08
Refactor and clean up for testing
16e7ba9d
Specify in comments
275a2017
Better error handling
c03e84f7
A million tests later
e4c7151d
More tests + err handling
284c08db
Require hub
5765b55e
More with remote code
0987db41
Clean up
1db6dcc5
Add a test for no_split
b0211904
Apply suggestions from code review
1ce3771e
Docstring
24d34351
Address some comments
8cea56e4
rm einops
e2abea0b
Let it err out
e8f07ff9
Adjust errs
39e1c785
Tests
b1cf0c71
Reduce test repeats
dd624b13
sgugger
approved these changes
on 2023-08-24
BenjaminBossan
approved these changes
on 2023-08-24
SunMarc
approved these changes
on 2023-08-24
Clean up borders
ee449d0c
Tip on 20%
4709df14
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub