camenduru f73ed17961 Allow converting Flax to PyTorch by adding a "from_flax" keyword (#1900) ...

* from_flax

* oops

* oops

* make style with pip install -e ".[dev]"

* oops

* now code quality happy 😋

* allow_patterns += FLAX_WEIGHTS_NAME

* Update src/diffusers/pipelines/pipeline_utils.py

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

* Update src/diffusers/pipelines/pipeline_utils.py

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

* Update src/diffusers/pipelines/pipeline_utils.py

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

* Update src/diffusers/pipelines/pipeline_utils.py

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

* Update src/diffusers/models/modeling_utils.py

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

* Update src/diffusers/pipelines/pipeline_utils.py

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

* for test

* bye bye is_flax_available()

* oops

* Update src/diffusers/models/modeling_pytorch_flax_utils.py

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* Update src/diffusers/models/modeling_pytorch_flax_utils.py

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* Update src/diffusers/models/modeling_pytorch_flax_utils.py

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* Update src/diffusers/models/modeling_utils.py

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* Update src/diffusers/models/modeling_utils.py

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* make style

* add test

* finihs

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

2023-01-12 20:00:35 +01:00

..

fixtures/custom_pipeline

Make repo structure consistent (#1862)

2022-12-30 11:51:08 +01:00

models

Make repo structure consistent (#1862)

2022-12-30 11:51:08 +01:00

pipelines

Fix slow tests (#1983)

2023-01-12 18:24:51 +01: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.12.0.dev0 (#1771)

2022-12-19 18:44:08 +01:00

test_layers_utils.py

Test ResnetBlock2D (#1850)

2023-01-04 22:57:32 +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 repo structure consistent (#1862)

2022-12-30 11:51:08 +01:00

test_outputs.py

Fix BaseOutput initialization from dict (#570)

2022-09-20 18:32:16 +02:00

test_pipelines_common.py

Fix MPS fast test warnings (#1744)

2022-12-17 22:57:30 +01: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_pipelines.py

Allow converting Flax to PyTorch by adding a "from_flax" keyword (#1900)

2023-01-12 20:00:35 +01:00

test_scheduler_flax.py

[Flax] Stateless schedulers, fixes and refactors (#1661)

2022-12-20 01:42:41 +01:00

test_scheduler.py

feat : add log-rho deis multistep scheduler (#1432)

2023-01-05 00:09:30 +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