Nicolas Patry d9227cf788 Adding use_safetensors argument to give more control to users (#2123) ...

* Adding `use_safetensors` argument to give more control to users

about which weights they use.

* Doc style.

* Rebased (not functional).

* Rebased and functional with tests.

* Style.

* Apply suggestions from code review

* Style.

* Addressing comments.

* Update tests/test_pipelines.py

Co-authored-by: Will Berman <wlbberman@gmail.com>

* Black ???

---------

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Will Berman <wlbberman@gmail.com>

2023-03-16 15:57:43 +01:00

..

fixtures/custom_pipeline

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

models

Adding use_safetensors argument to give more control to users (#2123)

2023-03-16 15:57:43 +01:00

pipelines

Add image_processor (#2617)

2023-03-15 07:55:49 -10:00

repo_utils

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

schedulers

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

__init__.py

fix issues with loading, add test for pipeline

2022-06-07 15:40:36 +02:00

conftest.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

pipeline_params.py

PipelineTesterMixin parameter configuration refactor (#2502)

2023-03-01 12:44:55 -08:00

test_config.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

test_ema.py

[Tests] fix: slow serialization test (#2678)

2023-03-15 22:30:21 +05:30

test_hub_utils.py

Remove modelcards dependency (#2050)

2023-01-20 16:39:42 +01:00

test_image_processor.py

Add image_processor (#2617)

2023-03-15 07:55:49 -10:00

test_layers_utils.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +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_modeling_common.py

[From pretrained] Speed-up loading from cache (#2515)

2023-03-10 11:56:10 +01:00

test_outputs.py

Replace flake8 with ruff and update black (#2279)

2023-02-07 23:46:23 +01:00

test_pipelines_common.py

[PipelineTesterMixin] Handle non-image outputs for attn slicing test (#2504)

2023-03-05 15:36:47 -08:00

test_pipelines_flax.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

test_pipelines_onnx_common.py

Reorganize pipeline tests (#963)

2022-10-24 16:34:01 +02:00

test_pipelines.py

Adding use_safetensors argument to give more control to users (#2123)

2023-03-16 15:57:43 +01:00

test_training.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

test_unet_2d_blocks.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

test_unet_blocks_common.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

test_utils.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00