Skip to content

Commit 573283e

Browse files
patil-surajpcuencapatrickvonplaten
authored
Add InstructPix2Pix pipeline (huggingface#2040)
* being pix2pix * ifx * cfg image_latents * fix some docstr * fix * fix * hack * fix * Apply suggestions from code review Co-authored-by: Pedro Cuenca <[email protected]> * add comments to explain the hack * move __call__ to the top * doc * remove height and width * remove depreications * fix doc str * quality * fast tests * chnage model id * fast tests * fix test * address Pedro's comments * copyright * Simple doc page. * Apply suggestions from code review * style * Remove import * address some review comments * Apply suggestions from code review Co-authored-by: Patrick von Platen <[email protected]> * style Co-authored-by: Pedro Cuenca <[email protected]> Co-authored-by: Patrick von Platen <[email protected]>
1 parent 27d2b2f commit 573283e

File tree

5 files changed

+637
-0
lines changed

5 files changed

+637
-0
lines changed

__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,7 @@
114114
StableDiffusionImg2ImgPipeline,
115115
StableDiffusionInpaintPipeline,
116116
StableDiffusionInpaintPipelineLegacy,
117+
StableDiffusionInstructPix2PixPipeline,
117118
StableDiffusionPipeline,
118119
StableDiffusionPipelineSafe,
119120
StableDiffusionUpscalePipeline,

pipelines/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,7 @@
5151
StableDiffusionImg2ImgPipeline,
5252
StableDiffusionInpaintPipeline,
5353
StableDiffusionInpaintPipelineLegacy,
54+
StableDiffusionInstructPix2PixPipeline,
5455
StableDiffusionPipeline,
5556
StableDiffusionUpscalePipeline,
5657
)

pipelines/stable_diffusion/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ class StableDiffusionPipelineOutput(BaseOutput):
4343
from .pipeline_stable_diffusion_img2img import StableDiffusionImg2ImgPipeline
4444
from .pipeline_stable_diffusion_inpaint import StableDiffusionInpaintPipeline
4545
from .pipeline_stable_diffusion_inpaint_legacy import StableDiffusionInpaintPipelineLegacy
46+
from .pipeline_stable_diffusion_instruct_pix2pix import StableDiffusionInstructPix2PixPipeline
4647
from .pipeline_stable_diffusion_upscale import StableDiffusionUpscalePipeline
4748
from .safety_checker import StableDiffusionSafetyChecker
4849

0 commit comments

Comments
 (0)