02d83c9ff1
Standardize fast pipeline tests with PipelineTestMixin ( #1526 )
...
* [WIP] Standardize fast pipeline tests with PipelineTestMixin
* refactor the sd tests a bit
* add more common tests
* add xformers
* add progressbar test
* cleanup
* upd fp16
* CycleDiffusionPipelineFastTests
* DanceDiffusionPipelineFastTests
* AltDiffusionPipelineFastTests
* StableDiffusion2PipelineFastTests
* StableDiffusion2InpaintPipelineFastTests
* StableDiffusionImageVariationPipelineFastTests
* StableDiffusionImg2ImgPipelineFastTests
* StableDiffusionInpaintPipelineFastTests
* remove unused mixins
* quality
* add missing inits
* try to fix mps tests
* fix mps tests
* add mps warmups
* skip for some pipelines
* style
* Update tests/test_pipelines_common.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
2022-12-06 18:35:30 +01:00
52eb0348e5
Standardize on using `image` argument in all pipelines ( #1361 )
...
* feat: switch core pipelines to use image arg
* test: update tests for core pipelines
* feat: switch examples to use image arg
* docs: update docs to use image arg
* style: format code using black and doc-builder
* fix: deprecate use of init_image in all pipelines
2022-12-01 16:55:22 +01:00
9e234d8048
handle fp16 in `UNet2DModel` ( #1216 )
...
* make sure fp16 runs well
* add fp16 test for superes
* Update src/diffusers/models/unet_2d.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* gen on cuda
* always run fast inferecne test on cpu
* run on cpu
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2022-11-23 11:13:34 +01:00
65d136e067
Add improved handling of pil ( #1309 )
...
* Better error message for transformers dummy
* [PIL] Better deprecation functionality
* up
2022-11-16 15:58:22 +01:00
7d0c272939
Match the generator device to the pipeline for DDPM and DDIM ( #1222 )
...
* Match the generator device to the pipeline for DDPM and DDIM
* style
* fix
* update values
* fix fast tests
* trigger slow tests
* deprecate
* last value fixes
* mps fixes
2022-11-09 23:00:23 +01:00
5a59f9b717
Add LDM Super Resolution pipeline ( #1116 )
...
* Add ldm super resolution pipeline
* style
* fix copies
* style
* fix doc
* Update src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* Update src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
* Update src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* Update src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* add doc
* address comments
* address comments
* fix doc
* minor
* add tests
* add tests
* load text encoder from subfolder
* fix test
* fix test
* style
* style
* handle mps latents
* unfix typo
* unfix typo
* Update tests/pipelines/latent_diffusion/test_latent_diffusion_superresolution.py
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* fix set_timesteps mps
* fix set_timesteps mps
* Update src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* Update src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* Update src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* Update src/diffusers/pipelines/latent_diffusion/pipeline_latent_diffusion_superresolution.py
Co-authored-by: Suraj Patil <surajp815@gmail.com>
* style
* test 64x64 instead of 256x256
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Suraj Patil <surajp815@gmail.com>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2022-11-09 13:42:16 +01:00
Anton Lozhkov