Anton Lozhkov d50e321745 Support SD2 attention slicing (#1397) ...

* Support SD2 attention slicing

* Support SD2 attention slicing

* Add more copies

* Use attn_num_head_channels in blocks

* fix-copies

* Update tests

* fix imports

2022-11-24 22:42:59 +01:00

..

fixtures/custom_pipeline

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

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

models

update unet2d (#1376)

2022-11-23 20:46:30 +01:00

pipelines

Support SD2 attention slicing (#1397)

2022-11-24 22:42:59 +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

Add Scheduler.from_pretrained and better scheduler changing (#1286)

2022-11-15 18:15:13 +01:00

test_layers_utils.py

VQ-diffusion (#658)

2022-11-03 16:10:28 +01:00

test_modeling_common_flax.py

Flax: Trickle down norm_num_groups (#789)

2022-10-11 20:05:10 +02:00

test_modeling_common.py

Add Scheduler.from_pretrained and better scheduler changing (#1286)

2022-11-15 18:15:13 +01:00

test_outputs.py

Fix BaseOutput initialization from dict (#570)

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

test_pipelines_common.py

Reorganize pipeline tests (#963)

2022-10-24 16:34:01 +02: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

Upscaling fixed (#1402)

2022-11-24 20:33:52 +01:00

test_scheduler_flax.py

Add Scheduler.from_pretrained and better scheduler changing (#1286)

2022-11-15 18:15:13 +01:00

test_scheduler.py

Add Scheduler.from_pretrained and better scheduler changing (#1286)

2022-11-15 18:15:13 +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 sample size (#1406)

2022-11-24 22:32:44 +01:00