Skip to content

[WIP] Add's Inpaint to Kolors #28

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
wants to merge 4 commits into
base: main
Choose a base branch
from
Draft

[WIP] Add's Inpaint to Kolors #28

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
d6bdd1a
add inpaint to kolors init
shauray8 Aug 8, 2024
73875c7
basic struct for inpaint
shauray8 Aug 8, 2024
fbdc7fb
working pipeline
shauray8 Aug 8, 2024
742f9fb
runs and tests
shauray8 Aug 12, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions src/diffusers/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -281,6 +281,7 @@
"KandinskyV22PriorEmb2EmbPipeline",
"KandinskyV22PriorPipeline",
"KolorsImg2ImgPipeline",
"KolorsInpaintPipeline",
"KolorsPipeline",
"LatentConsistencyModelImg2ImgPipeline",
"LatentConsistencyModelPipeline",
Expand Down Expand Up @@ -709,6 +710,7 @@
KandinskyV22PriorEmb2EmbPipeline,
KandinskyV22PriorPipeline,
KolorsImg2ImgPipeline,
KolorsInpaintPipeline,
KolorsPipeline,
LatentConsistencyModelImg2ImgPipeline,
LatentConsistencyModelPipeline,
Expand Down
2 changes: 2 additions & 0 deletions src/diffusers/pipelines/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -203,6 +203,7 @@
_import_structure["kolors"] = [
"KolorsPipeline",
"KolorsImg2ImgPipeline",
"KolorsInpaintPipeline",
"ChatGLMModel",
"ChatGLMTokenizer",
]
Expand Down Expand Up @@ -497,6 +498,7 @@
ChatGLMModel,
ChatGLMTokenizer,
KolorsImg2ImgPipeline,
KolorsInpaintPipeline,
KolorsPipeline,
)
from .latent_consistency_models import (
Expand Down
3 changes: 2 additions & 1 deletion src/diffusers/pipelines/auto_pipeline.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@
KandinskyV22Pipeline,
)
from .kandinsky3 import Kandinsky3Img2ImgPipeline, Kandinsky3Pipeline
from .kolors import KolorsImg2ImgPipeline, KolorsPipeline
from .kolors import KolorsImg2ImgPipeline, KolorsPipeline, KolorsInpaintPipeline
from .latent_consistency_models import LatentConsistencyModelImg2ImgPipeline, LatentConsistencyModelPipeline
from .pag import (
StableDiffusionControlNetPAGPipeline,
Expand Down Expand Up @@ -127,6 +127,7 @@
("if", IFInpaintingPipeline),
("kandinsky", KandinskyInpaintCombinedPipeline),
("kandinsky22", KandinskyV22InpaintCombinedPipeline),
("kolors", KolorsInpaintPipeline),
("stable-diffusion-controlnet", StableDiffusionControlNetInpaintPipeline),
("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetInpaintPipeline),
("stable-diffusion-xl-pag", StableDiffusionXLPAGInpaintPipeline),
Expand Down
2 changes: 2 additions & 0 deletions src/diffusers/pipelines/kolors/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
else:
_import_structure["pipeline_kolors"] = ["KolorsPipeline"]
_import_structure["pipeline_kolors_img2img"] = ["KolorsImg2ImgPipeline"]
_import_structure["pipeline_kolors_inpaint"] = ["KolorsInpaintPipeline"]
_import_structure["text_encoder"] = ["ChatGLMModel"]
_import_structure["tokenizer"] = ["ChatGLMTokenizer"]

Expand All @@ -36,6 +37,7 @@
else:
from .pipeline_kolors import KolorsPipeline
from .pipeline_kolors_img2img import KolorsImg2ImgPipeline
from .pipeline_kolors_inpaint import KolorsInpaintPipeline
from .text_encoder import ChatGLMModel
from .tokenizer import ChatGLMTokenizer

Expand Down
Loading