Anton Lozhkov c7b4acfb37 Add CPU offloading to UnCLIP (#1761) ...

* Add CPU offloading to UnCLIP

* use fp32 for testing the offload

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

..

fixtures/custom_pipeline

[Custom pipeline] Easier loading of local pipelines (#1327)

2022-11-17 16:00:26 +01:00

models

Re-add xformers enable to UNet2DCondition (#1627)

2022-12-09 14:05:38 +01:00

pipelines

Add CPU offloading to UnCLIP (#1761)

2022-12-19 14:44:08 +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

[Version] Bump to 0.11.0.dev0 (#1682)

2022-12-13 13:51:36 +01:00

test_layers_utils.py

Add paint by example (#1533)

2022-12-07 11:06:30 +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

Fix dtype model loading (#1449)

2022-11-30 11:31:50 +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

Nightly integration tests (#1664)

2022-12-16 18:51:11 +01:00

test_scheduler_flax.py

[Version] Bump to 0.11.0.dev0 (#1682)

2022-12-13 13:51:36 +01:00

test_scheduler.py

kakaobrain unCLIP (#1428)

2022-12-18 15:15:30 -08: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