Skip to content

Refactor MMDiT, add ImageToImage and Inpaint for SD3 #1909

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

Merged
merged 9 commits into from
Oct 8, 2024
Merged

Refactor MMDiT, add ImageToImage and Inpaint for SD3 #1909

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
e20ab7d
Refactor `MMDiT` and add `ImageToImage`
james77777778 Oct 5, 2024
a7cc7f2
Update model version
james77777778 Oct 5, 2024
da16e67
Fix minor bugs.
james77777778 Oct 5, 2024
8aa7388
Add `Inpaint` for SD3.
james77777778 Oct 6, 2024
c7749fb
Fix warnings of MMDiT.
james77777778 Oct 7, 2024
37c519b
Addcomment to Inpaint
james77777778 Oct 7, 2024
5ff2fa1
Simplify `MMDiT` implementation and info of `summary()`.
james77777778 Oct 7, 2024
eda16fc
Refactor `generate()` API of `TextToImage`, `ImageToImage` and `Inpai…
james77777778 Oct 8, 2024
023789e
Merge remote-tracking branch 'upstream/master' into refactor-mmdit
james77777778 Oct 8, 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
8 changes: 8 additions & 0 deletions keras_hub/api/models/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -180,6 +180,8 @@
from keras_hub.src.models.image_segmenter_preprocessor import (
ImageSegmenterPreprocessor,
)
from keras_hub.src.models.image_to_image import ImageToImage
from keras_hub.src.models.inpaint import Inpaint
from keras_hub.src.models.llama3.llama3_backbone import Llama3Backbone
from keras_hub.src.models.llama3.llama3_causal_lm import Llama3CausalLM
from keras_hub.src.models.llama3.llama3_causal_lm_preprocessor import (
Expand Down Expand Up @@ -270,6 +272,12 @@
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_backbone import (
StableDiffusion3Backbone,
)
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_image_to_image import (
StableDiffusion3ImageToImage,
)
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_inpaint import (
StableDiffusion3Inpaint,
)
from keras_hub.src.models.stable_diffusion_3.stable_diffusion_3_text_to_image import (
StableDiffusion3TextToImage,
)
Expand Down
Loading
Loading