diffusers/tests/pipelines/stable_diffusion_2
YiYi Xu 2e7a28652a
Attend and excite 2 (#2369)
* attend and excite pipeline

* update

update docstring example

remove visualization

remove the base class attention control

remove dependency on stable diffusion pipeline

always apply gaussian filter with default setting

remove run_standard_sd argument

hardcode attention_res and scale_range (related to step size)

Update docs/source/en/api/pipelines/stable_diffusion/attend_and_excite.mdx

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

Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py

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

Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py

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

Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py

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

Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py

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

Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py

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

Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py

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

Update tests/pipelines/stable_diffusion_2/test_stable_diffusion_attend_and_excite.py

Co-authored-by: Will Berman <wlbberman@gmail.com>

Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py

Co-authored-by: Will Berman <wlbberman@gmail.com>

Update src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_attend_and_excite.py

Co-authored-by: Will Berman <wlbberman@gmail.com>

revert test_float16_inference

revert change to the batch related tests

fix test_float16_inference

handle batch

remove the deprecation message

remove None check, step_size

remove debugging logging

add slow test

indices_to_alter -> indices

add check_input

* skip mps

* style

* Apply suggestions from code review

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

* indices -> token_indices
---------

Co-authored-by: evin <evinpinarornek@gmail.com>
Co-authored-by: yiyixuxu <yixu310@gmail,com>
Co-authored-by: Suraj Patil <surajp815@gmail.com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2023-02-16 11:15:54 -10:00
..
__init__.py Support SD2 attention slicing (#1397) 2022-11-24 22:42:59 +01:00
test_stable_diffusion.py `enable_model_cpu_offload` (#2285) 2023-02-16 19:06:36 +01:00
test_stable_diffusion_attend_and_excite.py Attend and excite 2 (#2369) 2023-02-16 11:15:54 -10:00
test_stable_diffusion_depth.py [Tests] Add MPS skip decorator (#2362) 2023-02-15 22:17:25 +01:00
test_stable_diffusion_flax.py Flax support for Stable Diffusion 2 (#1423) 2022-11-29 12:33:21 +01:00
test_stable_diffusion_flax_inpaint.py [Flax] Add Flax inpainting impl (#1966) 2023-01-17 10:42:04 +01:00
test_stable_diffusion_inpaint.py unet check length inputs (#2327) 2023-02-13 16:25:50 -08:00
test_stable_diffusion_latent_upscale.py schedulers add glide noising schedule (#2347) 2023-02-15 23:51:33 -08:00
test_stable_diffusion_upscale.py Replace flake8 with ruff and update black (#2279) 2023-02-07 23:46:23 +01:00
test_stable_diffusion_v_pred.py unet check length inputs (#2327) 2023-02-13 16:25:50 -08:00