Skip to content

Commit 5efb81f

Browse files
authored
Add ChromaInpaintPipeline (#12848)
* Add `ChromaInpaintPipeline` * Set `attention_mask` to `dtype=torch.bool` for `ChromaInpaintPipeline`. * Revert `.gitignore`.
1 parent b351be2 commit 5efb81f

File tree

6 files changed

+1224
-2
lines changed

6 files changed

+1224
-2
lines changed

docs/source/en/api/pipelines/chroma.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,3 +99,9 @@ image.save("chroma-single-file.png")
9999
[[autodoc]] ChromaImg2ImgPipeline
100100
- all
101101
- __call__
102+
103+
## ChromaInpaintPipeline
104+
105+
[[autodoc]] ChromaInpaintPipeline
106+
- all
107+
- __call__

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -460,6 +460,7 @@
460460
"BriaFiboPipeline",
461461
"BriaPipeline",
462462
"ChromaImg2ImgPipeline",
463+
"ChromaInpaintPipeline",
463464
"ChromaPipeline",
464465
"ChronoEditPipeline",
465466
"CLIPImageProjection",
@@ -1186,6 +1187,7 @@
11861187
BriaFiboPipeline,
11871188
BriaPipeline,
11881189
ChromaImg2ImgPipeline,
1190+
ChromaInpaintPipeline,
11891191
ChromaPipeline,
11901192
ChronoEditPipeline,
11911193
CLIPImageProjection,

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@
155155
"AudioLDM2UNet2DConditionModel",
156156
]
157157
_import_structure["blip_diffusion"] = ["BlipDiffusionPipeline"]
158-
_import_structure["chroma"] = ["ChromaPipeline", "ChromaImg2ImgPipeline"]
158+
_import_structure["chroma"] = ["ChromaPipeline", "ChromaImg2ImgPipeline", "ChromaInpaintPipeline"]
159159
_import_structure["cogvideo"] = [
160160
"CogVideoXPipeline",
161161
"CogVideoXImageToVideoPipeline",
@@ -598,7 +598,7 @@
598598
from .blip_diffusion import BlipDiffusionPipeline
599599
from .bria import BriaPipeline
600600
from .bria_fibo import BriaFiboPipeline
601-
from .chroma import ChromaImg2ImgPipeline, ChromaPipeline
601+
from .chroma import ChromaImg2ImgPipeline, ChromaInpaintPipeline, ChromaPipeline
602602
from .chronoedit import ChronoEditPipeline
603603
from .cogvideo import (
604604
CogVideoXFunControlPipeline,

src/diffusers/pipelines/chroma/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
else:
2525
_import_structure["pipeline_chroma"] = ["ChromaPipeline"]
2626
_import_structure["pipeline_chroma_img2img"] = ["ChromaImg2ImgPipeline"]
27+
_import_structure["pipeline_chroma_inpainting"] = ["ChromaInpaintPipeline"]
2728
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
2829
try:
2930
if not (is_transformers_available() and is_torch_available()):
@@ -33,6 +34,7 @@
3334
else:
3435
from .pipeline_chroma import ChromaPipeline
3536
from .pipeline_chroma_img2img import ChromaImg2ImgPipeline
37+
from .pipeline_chroma_inpainting import ChromaInpaintPipeline
3638
else:
3739
import sys
3840

0 commit comments

Comments
 (0)