Suraj Patil
5383188c7e
StableDiffusionDepth2ImgPipeline ( #1531 )
...
* begin depth pipeline
* add depth estimation model
* fix prepare_depth_mask
* add a comment about autocast
* copied from, quality, cleanup
* begin tests
* handle tensors
* norm image tensor
* fix batch size
* fix tests
* fix enable_sequential_cpu_offload
* fix save load
* fix test_save_load_float16
* fix test_save_load_optional_components
* fix test_float16_inference
* fix test_cpu_offload_forward_pass
* fix test_dict_tuple_outputs_equivalent
* up
* fix fast tests
* fix test_stable_diffusion_img2img_multiple_init_images
* fix few more fast tests
* don't use device map for DPT
* fix test_stable_diffusion_pipeline_with_sequential_cpu_offloading
* accept external depth maps
* prepare_depth_mask -> prepare_depth_map
* fix file name
* fix file name
* quality
* check transformers version
* fix test names
* use skipif
* fix import
* add docs
* skip tests on mps
* correct version
* uP
* Update docs/source/api/pipelines/stable_diffusion_2.mdx
* fix fix-copies
* fix fix-copies
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: anton- <anton@huggingface.co>
2022-12-08 18:25:12 +01:00
Anton Lozhkov
dc87f526d4
Fix common tests for FP16 ( #1588 )
...
* Fix common tests for FP16
* revert
2022-12-07 14:09:51 +01:00
Patrick von Platen
896c98a2ae
Add paint by example ( #1533 )
...
* add paint by example
* mkae loading possibel
* up
* Update src/diffusers/models/attention.py
* up
* finalize weight structure
* make example work
* make it work
* up
* up
* fix
* del
* add
* update
* Apply suggestions from code review
* correct transformer 2d
* finish
* up
* up
* up
* up
* fix
* Apply suggestions from code review
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
* Apply suggestions from code review
* up
* finish
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
2022-12-07 11:06:30 +01:00
Anton Lozhkov
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
Anton Lozhkov
2c82e0c4eb
Reorganize pipeline tests ( #963 )
...
* Reorganize pipeline tests
* fix vq
2022-10-24 16:34:01 +02:00