tests/test_modeling_common.py · 31565ff0fd827c7ffa707d8b99cb2abf7fa7f6bb · chenpangpang / transformers

Poc to use safetensors (#19175) · 3e2dd7f9

Sylvain Gugger authored Sep 30, 2022



* Poc to use safetensors

* Typo

* Final version

* Add tests

* Save with the right name!

* Update tests/test_modeling_common.py
Co-authored-by: Julien Chaumond <julien@huggingface.co>

* Support for sharded checkpoints

* Test from Hub part 1

* Test from hub part 2

* Fix regular checkpoint sharding

* Bump for fixes
Co-authored-by: Julien Chaumond <julien@huggingface.co>

3e2dd7f9

test_modeling_common.py 138 KB

Replace test_modeling_common.py