diffusers/tests
Matthieu Bizien ae368e42d2
[Proposal] Support saving to safetensors (#1494)
* Add parameter safe_serialization to DiffusionPipeline.save_pretrained

* Add option safe_serialization on ModelMixin.save_pretrained

* Add test test_save_safe_serialization

* Black

* Re-trigger the CI

* Fix doc-builder

* Validate files are saved as safetensor in test_save_safe_serialization
2022-12-02 18:33:16 +01:00
..
fixtures/custom_pipeline [Custom pipeline] Easier loading of local pipelines (#1327) 2022-11-17 16:00:26 +01:00
models Fix dtype model loading (#1449) 2022-11-30 11:31:50 +01:00
pipelines fix tests 2022-12-02 17:27:58 +00:00
repo_utils Move accelerate to a soft-dependency (#1134) 2022-11-04 14:58:52 +01:00
__init__.py fix issues with loading, add test for pipeline 2022-06-07 15:40:36 +02:00
conftest.py [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +02:00
test_config.py Bump to 0.10.0.dev0 + deprecations (#1490) 2022-11-30 15:27:56 +01:00
test_layers_utils.py VQ-diffusion (#658) 2022-11-03 16:10:28 +01:00
test_modeling_common.py Fix dtype model loading (#1449) 2022-11-30 11:31:50 +01:00
test_modeling_common_flax.py Allow to set config params directly in init (#1419) 2022-11-25 15:07:09 +01:00
test_outputs.py Fix BaseOutput initialization from dict (#570) 2022-09-20 18:32:16 +02:00
test_pipelines.py [Proposal] Support saving to safetensors (#1494) 2022-12-02 18:33:16 +01:00
test_pipelines_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_pipelines_flax.py Make height and width optional (#1401) 2022-11-24 18:23:59 +01:00
test_pipelines_onnx_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_scheduler.py Finalize 2nd order schedulers (#1503) 2022-12-02 16:38:35 +01:00
test_scheduler_flax.py Bump to 0.10.0.dev0 + deprecations (#1490) 2022-11-30 15:27:56 +01:00
test_training.py [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +02:00
test_utils.py [Deprecate] Correct stacklevel (#1483) 2022-12-01 16:28:10 +01:00