huggingface_hub
Making wrapper tensor subclass to work in serialization
#2440
Merged
Merged
Making wrapper tensor subclass to work in serialization #2440
Wauplin
merged
18 commits into
huggingface:main
from
jerryzh168:non-safetensor-ser
jerryzh168
changed the title
Making wrapper tensor subclass to work in huggingface_hub serializati…
a9911b8d
add tests
4ebe8aa9
update signature to include new changes for tensor subclass
0a1cc964
add torch version checks and move around import
8fd26b75
more fixes
b8890190
tested with torch 2.0.0 and 2.5.0
c91fe184
Merge branch 'main' into non-safetensor-ser
83f6884e
Merge branch 'main' into non-safetensor-ser
a02657d8
Wauplin
commented
on 2024-08-20
remove torch_version_at_least from _torch.py
94b38853
simplify code for checking if tensor subclass is available or not
1b25148e
minor fix
ca3c2280
addressing comments and run tests with torch 2.4.0
122a68a8
jerryzh168
commented
on 2024-08-24
Merge branch 'main' into non-safetensor-ser
3c796075
some linting
7fb689bd
add test_split_torch_state_dict_into_shards for tensor subclass state…
45cb98ad
lint
1486253d
style
7850025e
quality
473c317d
Wauplin
approved these changes
on 2024-08-30
Login to write a write a comment.
Login via GitHub