Pedro Cuenca 2777264ee8 enable_model_cpu_offload (#2285) ...

* enable_model_offload PoC

It's surprisingly more involved than expected, see comments in the PR.

* Rename final_offload_hook

* Invoke the vae forward hook manually.

* Completely remove decoder.

* Style

* apply_forward_hook decorator

* Rename method.

* Style

* Copy enable_model_cpu_offload

* Fix copies.

* Remove comment.

* Fix copies

* Missing import

* Fix doc-builder style.

* Merge main and fix again.

* Add docs

* Fix docs.

* Add a couple of tests.

* style

2023-02-16 19:06:36 +01:00

..

fixtures/custom_pipeline

Make repo structure consistent (#1862)

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

models

Fix running LoRA with xformers (#2286)

2023-02-13 11:58:18 +01:00

pipelines

enable_model_cpu_offload (#2285)

2023-02-16 19:06:36 +01:00

repo_utils

Replace flake8 with ruff and update black (#2279)

2023-02-07 23:46:23 +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 version] 0.13.0dev0 & Deprecate predict_epsilon (#2109)

2023-01-25 17:59:02 +01:00

test_hub_utils.py

Remove modelcards dependency (#2050)

2023-01-20 16:39:42 +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

[Variant] Add "variant" as input kwarg so to have better UX when downloading no_ema or fp16 weights (#2305)

2023-02-16 11:02:58 +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

[Pipelines] Adds pix2pix zero (#2334)

2023-02-16 11:20:38 +01:00

test_pipelines_flax.py

Replace flake8 with ruff and update black (#2279)

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

test_pipelines_onnx_common.py

Reorganize pipeline tests (#963)

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

test_pipelines.py

[Variant] Add "variant" as input kwarg so to have better UX when downloading no_ema or fp16 weights (#2305)

2023-02-16 11:02:58 +01:00

test_scheduler_flax.py

[Bump version] 0.13.0dev0 & Deprecate predict_epsilon (#2109)

2023-01-25 17:59:02 +01:00

test_scheduler.py

[Bug]: fix DDPM scheduler arbitrary infer steps count. (#2076)

2023-01-31 09:13:26 +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_unet_2d_blocks.py

Add tests for 2D UNet blocks (#1945)

2023-01-16 12:53:05 +01:00

test_unet_blocks_common.py

Add tests for 2D UNet blocks (#1945)

2023-01-16 12:53:05 +01:00

test_utils.py

[Deprecate] Correct stacklevel (#1483)

2022-12-01 16:28:10 +01:00