Skip to content

Commit e2c7fb5

Browse files
committed
Add ChromaInpaintPipeline
1 parent 5e48f46 commit e2c7fb5

File tree

7 files changed

+1249
-3
lines changed

7 files changed

+1249
-3
lines changed

.gitignore

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,4 +178,9 @@ tags
178178
.ruff_cache
179179

180180
# wandb
181-
wandb
181+
wandb
182+
183+
# pixi environments
184+
.pixi/*
185+
!.pixi/config.toml
186+
pixi.toml

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
@@ -451,6 +451,7 @@
451451
"BriaFiboPipeline",
452452
"BriaPipeline",
453453
"ChromaImg2ImgPipeline",
454+
"ChromaInpaintPipeline",
454455
"ChromaPipeline",
455456
"ChronoEditPipeline",
456457
"CLIPImageProjection",
@@ -1159,6 +1160,7 @@
11591160
BriaFiboPipeline,
11601161
BriaPipeline,
11611162
ChromaImg2ImgPipeline,
1163+
ChromaInpaintPipeline,
11621164
ChromaPipeline,
11631165
ChronoEditPipeline,
11641166
CLIPImageProjection,

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@
154154
"AudioLDM2UNet2DConditionModel",
155155
]
156156
_import_structure["blip_diffusion"] = ["BlipDiffusionPipeline"]
157-
_import_structure["chroma"] = ["ChromaPipeline", "ChromaImg2ImgPipeline"]
157+
_import_structure["chroma"] = ["ChromaPipeline", "ChromaImg2ImgPipeline", "ChromaInpaintPipeline"]
158158
_import_structure["cogvideo"] = [
159159
"CogVideoXPipeline",
160160
"CogVideoXImageToVideoPipeline",
@@ -585,7 +585,7 @@
585585
from .blip_diffusion import BlipDiffusionPipeline
586586
from .bria import BriaPipeline
587587
from .bria_fibo import BriaFiboPipeline
588-
from .chroma import ChromaImg2ImgPipeline, ChromaPipeline
588+
from .chroma import ChromaImg2ImgPipeline, ChromaInpaintPipeline, ChromaPipeline
589589
from .chronoedit import ChronoEditPipeline
590590
from .cogvideo import (
591591
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)