PR #1494 [Proposal] Support saving to safetensors

[Proposal] Support saving to safetensors #1494

patrickvonplaten merged 9 commits into huggingface:main from MatthieuBizien:save_safetensors

Add parameter safe_serialization to DiffusionPipeline.save_pretrained

e1c00686

Add option safe_serialization on ModelMixin.save_pretrained

07a8da16

Add test test_save_safe_serialization

1a66a2d0

Black

06c834ff

Merge branch 'main' into save_safetensors

334a093b

MatthieuBizien marked this pull request as ready for review 3 years ago

Re-trigger the CI

c1923705

Fix doc-builder

222c1191

Narsil approved these changes on 2022-12-01

Validate files are saved as safetensor in test_save_safe_serialization

8c12a559

Merge branch 'main' into save_safetensors

fe8df3a8

Narsil approved these changes on 2022-12-02

patrickvonplaten merged ae368e42 into main 3 years ago

Reviewers

Narsil

Assignees

No one assigned

Labels

None yet

Milestone

No milestone