burn/crates/burn-import/safetensors-tests
Brandon Wand eb57d7aaa1
Support importing safetensors format (#2721)
* add support for safetensors in pytorch reader

* update book

* support safetensors format, for it's own crate

* remove some duplicate code and depend on pytorch adapter

* Fix clippy error

* Fix clippy errors

* Update doc strings

* Move common types/functions into common module

* Refactor and add adapter option

* Update doc strings

* Updated example for importing pt and safetensors weights

* Add book section on safetensors

* Mention Safetensors in the main README file

* Remove dead code

* SafeTensors -> Safetensors type

* SafeTensors -> Safetensors in docs/messages

* SafeTensors -> Safetensors in book

* SafeTensors -> Safetensors in example

* SafeTensors -> Safetensors in readme

* Removed TensorFlow

* Removed excess documentation

* Update pytorch-model to look the formatting similar to safetensors md

* Update pytorch-model.md

* Removed pytorch copied tests and replaced with one multilayer test

* Update README.md

Co-authored-by: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>

* Update burn-book/src/import/README.md

Co-authored-by: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>

* Update burn-book/src/import/README.md

Co-authored-by: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>

* Shorter example

* Remove redundant comment

* Removed extra wording

* Replace unwrap_or_else with expect

* Removed imports for pytorch doc consistency

* Removed imports

* Fix formating

---------

Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
Co-authored-by: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>
2025-05-06 12:51:08 -04:00
..
src Support importing safetensors format (#2721) 2025-05-06 12:51:08 -04:00
tests Support importing safetensors format (#2721) 2025-05-06 12:51:08 -04:00
Cargo.toml Support importing safetensors format (#2721) 2025-05-06 12:51:08 -04:00