diffusers/tests
Revist d38c804320
feat: add repaint (#974)
* feat: add repaint

* fix: fix quality check with `make fix-copies`

* fix: remove old unnecessary arg

* chore: change default to DDPM (looks better in experiments)

* ".to(device)" changed to "device="

Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>

* make generator device-specific

Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>

* make generator device-specific and change shape

Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>

* fix: add preprocessing for image and mask

Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>

* fix: update test

Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>

* Update src/diffusers/pipelines/repaint/pipeline_repaint.py

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

* Add docs and examples

* Fix toctree

Co-authored-by: fja <fja@zurich.ibm.com>
Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Anton Lozhkov <anton@huggingface.co>
2022-11-03 15:42:46 +01:00
..
fixtures/custom_pipeline Custome Pipelines (#744) 2022-10-06 16:54:02 +02:00
models Integration tests precision improvement for inpainting (#1052) 2022-11-02 11:47:26 +01:00
pipelines feat: add repaint (#974) 2022-11-03 15:42:46 +01:00
repo_utils Introduce the copy mechanism (#924) 2022-10-20 20:26:03 +02: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 Fix hub-dependent tests for PRs (#1119) 2022-11-03 15:24:32 +01:00
test_layers_utils.py Add Apple M1 tests (#796) 2022-10-17 20:27:30 +02:00
test_modeling_common.py Add Apple M1 tests (#796) 2022-10-17 20:27:30 +02:00
test_modeling_common_flax.py Flax: Trickle down `norm_num_groups` (#789) 2022-10-11 20:05:10 +02:00
test_outputs.py Fix BaseOutput initialization from dict (#570) 2022-09-20 18:32:16 +02:00
test_pipelines.py [Loading] Ignore unneeded files (#1107) 2022-11-02 19:20:42 +01:00
test_pipelines_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_pipelines_flax.py [Loading] Ignore unneeded files (#1107) 2022-11-02 19:20:42 +01:00
test_pipelines_onnx_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_scheduler.py k-diffusion-euler (#1019) 2022-10-31 16:20:38 +01:00
test_scheduler_flax.py [CI] Framework and hardware-specific CI tests (#997) 2022-11-02 14:07:07 +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 [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +02:00