[feat] JoyAI-JoyImage-Edit support #13444
yiyixuxu
merged
46 commits into
huggingface:main
from
Moran232:joyimage_edit
[feat] JoyAI-JoyImage-Edit support
54afa650
dg845
commented
on 2026-04-14
dg845
commented
on 2026-04-14
dg845
commented
on 2026-04-14
dg845
commented
on 2026-04-14
dg845
commented
on 2026-04-14
dg845
commented
on 2026-04-14
dg845
commented
on 2026-04-14
dg845
commented
on 2026-04-14
dg845
commented
on 2026-04-14
dg845
commented
on 2026-04-14
[fix] remove rearrange
84597596
[refactor] two pass when do cfg
e6e6df53
[refactor] remove repa, use wantimetextembeding, refactor modulate code
f557113a
[refactor] Joyimage Attention refactor
d397b681
remove vae tiling and autocast
9d78e4e1
dg845
commented
on 2026-04-16
dg845
commented
on 2026-04-16
dg845
commented
on 2026-04-16
dg845
commented
on 2026-04-16
dg845
commented
on 2026-04-16
[fix] remove einops from setup.py
cc9d1343
[refactor] Refactor JoyImageEditPipeline to use explicit arguments in…
001f7d34
[fix] remove deprecated method decode_latents
657b4b07
[refactor] refactor the image pre-processing logic into a separate Va…
19027dd0
[refactor] add JoyImageAttention to align with Attention + AttnProces…
0a06f33f
[refactor] simplify bucket logic in JoyImageEditImageProcessor by rep…
02d947c4
[fix] remove leftover training-only parameters
2b7fb866
[fix] add layerwise casting and fp32 module patterns to JoyImageTrans…
79c48a87
[test] add JoyImageEditPipeline fast tests and JoyImageEditTransforme…
98cee973
[fix] fix some pipeline args to support batch inference
a716050a
[fix] duplicate images to match batch size when fewer images than pro…
320bde7b
[fix] remove no longer used config parameters
3ed6ca9b
Merge branch 'main' into joyimage_edit
92f4d852
Apply style fixes
261613be
dg845
commented
on 2026-04-28
dg845
commented
on 2026-04-28
dg845
commented
on 2026-04-28
dg845
commented
on 2026-04-28
dg845
commented
on 2026-04-28
dg845
commented
on 2026-04-28
dg845
commented
on 2026-04-28
[fix] remove unused dataclass and rewrite helpers as inline functions
f364da3d
[fix] make dummy objects for JoyImageEdit
c7bd2849
[fix] allow test_torch_compile_repeated_blocks to pass
e45e1ad7
[fix] add examples on JoyImageEditPipeline
aeaa3345
fix code style issues with ruff and black
ce4a3d91
Apply style fixes
82e5cd2c
[fix] change default num_inference_steps to 40
844f3f15
[fix] use forward hook to extract pre-norm hidden states for transfor…
fd29a731
[fix] change the assert to ValueError in pipeline
3a6b658b
[fix] rename JoyImageTransformer3DModel to JoyImageEditTransformer3DM…
76c1647b
[fix] support gradient checkpointing
deb5d4f6
[refactor] simplify RoPE utilities, inline helpers, copy WanTimeTextI…
d8c7c0e4
[fix] remove _get_text_encoder_ckpt and qwen_processor
203494ec
[fix] change nn.RMSNorm to FP32LayerNorm
8408f554
[fix] small fixes for suggestions given by Claude
88579cc9
Merge branch 'main' into joyimage_edit
ea529350
dg845
commented
on 2026-04-30
dg845
commented
on 2026-04-30
dg845
commented
on 2026-04-30
dg845
commented
on 2026-04-30
dg845
commented
on 2026-04-30
dg845
approved these changes
on 2026-04-30
[refactor] build model using from _pretained instead of config
9d9ef527
[refactor] auto-wrap prompt and support text-to-image in JoyImage Edi…
87b5383d
make style, make quality and make fix-copies
cf61b0a6
[test] small fix to use vocab_size=1024
dbfbb597
Merge branch 'main' into joyimage_edit
756904e0
yiyixuxu
commented
on 2026-05-01
Merge branch 'main' into joyimage_edit
dc5a0a2f
[refactor] separate encode_prompt_multiple_images from encode_prompt,…
63baf43e
dg845
commented
on 2026-05-04
dg845
commented
on 2026-05-04
dg845
commented
on 2026-05-04
[test] fix CI: use strict=False for xfail and add @require_torch_acce…
e8c4db7b
yiyixuxu
approved these changes
on 2026-05-05
[refactor] separate image_latents from latents in prepare_latents to …
d54fed62
make style
23ba73e1
Reviewers
Assignees
No one assigned
Labels
models
tests
size/L
utils
pipelines
fixes-issue
Milestone
No milestone
Login to write a write a comment.
Login via GitHub