Patrick von Platen 5b6582cf73 [Model offload] Add nice warning (#2543) ...

* [Model offload] Add nice warning

* Treat sequential and model offload differently.

Sequential raises an error because the operation would fail with a
cryptic warning later.

* Forcibly move to cpu when offloading.

* make style

* one more fix

* make fix-copies

* up

---------

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

2023-03-03 16:44:10 +01:00

..

fixtures/custom_pipeline

[Copyright] 2023 (#2524)

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

models

[Copyright] 2023 (#2524)

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

pipelines

8k Stable Diffusion with tiled VAE (#1441)

2023-03-02 17:42:32 +01:00

repo_utils

[Copyright] 2023 (#2524)

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

Remove modelcards dependency (#2050)

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

[Copyright] 2023 (#2524)

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

Add a ControlNet model & pipeline (#2407)

2023-03-02 15:34:07 +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

[Model offload] Add nice warning (#2543)

2023-03-03 16:44:10 +01:00

test_scheduler_flax.py

[Copyright] 2023 (#2524)

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

test_scheduler.py

[Copyright] 2023 (#2524)

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