Pedro Cuenca 81125d8499 Make dynamo wrapped modules work with save_pretrained (#2726) ...

* Workaround for saving dynamo-wrapped models.

* Accept suggestion from code review

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

* Apply workaround when overriding pipeline components.

* Ensure the correct config.json is saved to disk.

Instead of the dynamo class.

* Save correct module (not compiled one)

* Add test

* style

* fix docstrings

* Go back to using string comparisons.

PyTorch CPU does not have _dynamo.

* Simple test for save_pretrained of compiled models.

* Helper function to test whether module is compiled.

---------

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

2023-03-28 09:03:21 +02:00

..

fixtures

Music Spectrogram diffusion pipeline (#1044)

2023-03-23 14:06:17 +01:00

models

Helper function to disable custom attention processors (#2791)

2023-03-27 20:31:19 +02:00

pipelines

Helper function to disable custom attention processors (#2791)

2023-03-27 20:31:19 +02: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

Add AudioLDM (#2232)

2023-03-23 19:00:21 +01:00

test_config.py

[Copyright] 2023 (#2524)

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

test_ema.py

[Tests] Correct PT2 (#2724)

2023-03-18 18:38:04 +01:00

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

Make dynamo wrapped modules work with save_pretrained (#2726)

2023-03-28 09:03:21 +02:00

test_outputs.py

Replace flake8 with ruff and update black (#2279)

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

test_pipelines_common.py

mps: remove warmup passes (#2771)

2023-03-22 19:29:27 +01: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

Make dynamo wrapped modules work with save_pretrained (#2726)

2023-03-28 09:03:21 +02:00

test_training.py

[Copyright] 2023 (#2524)

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

test_unet_2d_blocks.py

Fix mps tests on torch 2.0 (#2766)

2023-03-21 15:19:31 +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