StableDiffusionImageVariationPipeline (#1365) · feisan/diffusers@0eb507f · GitHub
Skip to content

Commit 0eb507f

Browse files
patil-surajpcuenca
andauthored
StableDiffusionImageVariationPipeline (huggingface#1365)
* add StableDiffusionImageVariationPipeline * add ini init * use CLIPVisionModelWithProjection * fix _encode_image * add copied from * fix copies * add doc * handle tensor in _encode_image * add tests * correct model_id * remove copied from in enable_sequential_cpu_offload * fix tests * make slow tests pass * update slow tests * use temp model for now * fix test_stable_diffusion_img_variation_intermediate_state * fix test_stable_diffusion_img_variation_intermediate_state * check for torch.Tensor * quality * fix name * fix slow tests * install transformers from source * fix install * fix install * Apply suggestions from code review Co-authored-by: Pedro Cuenca <pedro@huggingface.co> * input_image -> image * remove deprication warnings * fix test_stable_diffusion_img_variation_multiple_images * make flake happy Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
1 parent 9e234d8 commit 0eb507f

9 files changed

Lines changed: 890 additions & 0 deletions

File tree

‎.github/workflows/pr_tests.yml‎

Lines changed: 2 additions & 0 deletions

‎.github/workflows/push_tests.yml‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ jobs:
6262
run: |
6363
python -m pip install -e .[quality,test]
6464
python -m pip install git+https://github.com/huggingface/accelerate
65+
python -m pip install -U git+https://github.com/huggingface/transformers
6566
6667
- name: Environment
6768
run: |
@@ -131,6 +132,7 @@ jobs:
131132
run: |
132133
python -m pip install -e .[quality,test,training]
133134
python -m pip install git+https://github.com/huggingface/accelerate
135+
python -m pip install -U git+https://github.com/huggingface/transformers
134136
135137
- name: Environment
136138
run: |

‎docs/source/api/pipelines/stable_diffusion.mdx‎

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,3 +88,10 @@ If you want to use all possible use cases in a single `DiffusionPipeline` you ca
8888
- __call__
8989
- enable_attention_slicing
9090
- disable_attention_slicing
91+
92+
93+
## StableDiffusionImageVariationPipeline
94+
[[autodoc]] StableDiffusionImageVariationPipeline
95+
- __call__
96+
- enable_attention_slicing
97+
- disable_attention_slicing

‎src/diffusers/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@
6969
AltDiffusionPipeline,
7070
CycleDiffusionPipeline,
7171
LDMTextToImagePipeline,
72+
StableDiffusionImageVariationPipeline,
7273
StableDiffusionImg2ImgPipeline,
7374
StableDiffusionInpaintPipeline,
7475
StableDiffusionInpaintPipelineLegacy,

‎src/diffusers/pipelines/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from .latent_diffusion import LDMTextToImagePipeline
2020
from .stable_diffusion import (
2121
CycleDiffusionPipeline,
22+
StableDiffusionImageVariationPipeline,
2223
StableDiffusionImg2ImgPipeline,
2324
StableDiffusionInpaintPipeline,
2425
StableDiffusionInpaintPipelineLegacy,

‎src/diffusers/pipelines/stable_diffusion/__init__.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ class StableDiffusionPipelineOutput(BaseOutput):
3030
if is_transformers_available() and is_torch_available():
3131
from .pipeline_cycle_diffusion import CycleDiffusionPipeline
3232
from .pipeline_stable_diffusion import StableDiffusionPipeline
33+
from .pipeline_stable_diffusion_image_variation import StableDiffusionImageVariationPipeline
3334
from .pipeline_stable_diffusion_img2img import StableDiffusionImg2ImgPipeline
3435
from .pipeline_stable_diffusion_inpaint import StableDiffusionInpaintPipeline
3536
from .pipeline_stable_diffusion_inpaint_legacy import StableDiffusionInpaintPipelineLegacy

‎src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py‎

Lines changed: 437 additions & 0 deletions
Large diffs are not rendered by default.

‎src/diffusers/utils/dummy_torch_and_transformers_objects.py‎

Lines changed: 15 additions & 0 deletions

0 commit comments

Comments
 (0)