Making wrapper tensor subclass to work in serialization#2440

Merged

Wauplin merged 18 commits intohuggingface:mainfrom jerryzh168:non-safetensor-ser

Aug 30, 2024

+176-11

Commits on Aug 9, 2024

Commits on Aug 10, 2024

update signature to include new changes for tensor subclass
jerryzh168
committed

Commits on Aug 19, 2024

Commits on Aug 22, 2024

Commits on Aug 23, 2024

addressing comments and run tests with torch 2.4.0
jerryzh168
committed

Commits on Aug 26, 2024

Commits on Aug 29, 2024

add test_split_torch_state_dict_into_shards for tensor subclass state dict
jerryzh168
committed

Commits on Aug 30, 2024