Skip to content

Add save in safetensors format #784

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 33 commits into from
Jun 30, 2025
Merged

Add save in safetensors format #784

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
33 commits
Select commit Hold shift + click to select a range
8a2fc15
Add save in safetensors format
Jun 9, 2025
bdf275a
Add safetensors to reqs
Jun 9, 2025
b000c74
Add loading safetensors
Jun 9, 2025
a86d4e4
Add logic for onnx weights
Jun 9, 2025
1192a21
Update load
Jun 10, 2025
d0de633
Update save method
Jun 10, 2025
e3f15db
Add pickle_metadata flag
Jun 10, 2025
ba767b0
Add return for onnx load
Jun 10, 2025
f7cc818
Add preserved_dict for safetensors
Jun 10, 2025
d5bba46
Update batchflow/models/torch/base.py
igor-iusupov Jun 10, 2025
e515c65
Update saving extensions
Jun 10, 2025
b491290
Update load model for safetensors
Jun 10, 2025
3d9392c
Small update
Jun 10, 2025
5304560
Update path logic. Make path optional
Jun 10, 2025
7305ea9
Fix for linter
Jun 10, 2025
7b5217a
Remove all use_xxx and add format=name
Jun 11, 2025
9476e3c
AssertError -> ValueError
Jun 11, 2025
88f7378
Format -> fmt
Jun 11, 2025
23d4d94
Add fmt to load
Jun 11, 2025
fcbb4b7
Update saving model name
Jun 11, 2025
5b0f63d
Small update
Jun 11, 2025
53d6cc9
Update batchflow/models/torch/base.py
igor-iusupov Jun 11, 2025
19b135d
Update docstring
Jun 11, 2025
8ed42de
Remove device from load_file func
Jun 11, 2025
ed57f96
Make fmt optional
Jun 16, 2025
169f58a
Update model init
Jun 16, 2025
429d462
Fix model_to_device
AlexeyKozhevin Jun 27, 2025
f632dea
Mark test_redirect_stdout as slow
AlexeyKozhevin Jun 27, 2025
2a8fce9
Refactoring
AlexeyKozhevin Jun 27, 2025
c298bd1
Fix names collisions
AlexeyKozhevin Jun 27, 2025
eb306ba
Update tests and load/save
AlexeyKozhevin Jun 30, 2025
be86d99
Change logic
AlexeyKozhevin Jun 30, 2025
739af96
Minor changes
AlexeyKozhevin Jun 30, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading