Chroma Pipeline #11698
DN6
merged
117 commits into
huggingface:main
from
Ednaordinary:chroma
working state from hameerabbasi and iddl
ff0b9a3c
working state form hameerabbasi and iddl (transformer)
3c2865c5
working state (normalization)
e271af94
working state (embeddings)
15f2bd5c
add chroma loader
32e6a006
add chroma to mappings
bc36a0d8
add chroma to transformer init
33ea0b65
take out variant stuff
22ecd19f
get decently far in changing variant stuff
b0df9691
add chroma init
c8cbb316
make chroma output class
32659236
add chroma transformer to dummy tp
74002788
add chroma to init
c22930d7
add chroma to init
4e698b10
fix single file
5eb4b822
update
f0c75b6b
update
6441e70d
add chroma to auto pipeline
a6f231c7
add chroma to pipeline init
7445cf42
change to chroma transformer
af918c89
take out variant from blocks
2fcc75a6
swap embedder location
0b027a24
remove prompt_2
6c0aed14
work on swapping text encoders
f190c02a
remove mask function
38429ffc
dont modify mask (for now)
7c75d8e9
wrap attn mask
c9b46af6
no attn mask (can't get it to work)
146255ab
remove pooled prompt embeds
3309ffef
change to my own unpooled embeddeer
77b429ed
fix load
df7fde7a
take pooled projections out of transformer
68f771bf
Merge branch 'main' into chroma
a3b6697b
ensure correct dtype for chroma embeddings
f783f388
update
f6de1afc
use dn6 attn mask + fix true_cfg_scale
ab794217
use chroma pipeline output
442f77a2
use DN6 embeddings
e69d7309
remove guidance
01bc0dcc
remove guidance embed (pipeline)
e31c9486
remove guidance from embeddings
406ab3b1
don't return length
1bd8fdfc
Merge branch 'main' into chroma
2d57f3db
dont change dtype
3e2452de
remove unused stuff, fix up docs
1efa772f
add chroma autodoc
619921ca
add .md (oops)
f821f2ad
initial chroma docs
b0cf6803
undo don't change dtype
0c5eb447
undo arxiv change
42c0e8ec
fix hf papers regression in more places
da846d1f
Update docs/source/en/api/pipelines/chroma.md
18327cb5
do_cfg -> self.do_classifier_free_guidance
3f39b1a7
Update docs/source/en/api/models/chroma_transformer.md
a93e64d6
Update chroma.md
3e36a21c
Move chroma layers into transformer
a1fac68a
Remove pruned AdaLayerNorms
1442c978
Add chroma fast tests
03fbd520
(untested) batch cond and uncond
bedb3208
Add # Copied from for shift
fe5af79a
Update # Copied from statements
6a0db55a
update norm imports
abf8a33a
Revert cond + uncond batching
7235805e
Add transformer tests
15ca813e
move chroma test (oops)
f8d4a1a7
chroma init
c8d6aef9
fix chroma pipeline fast tests
cfd5b340
Update src/diffusers/models/transformers/transformer_chroma.py
2347d53f
Move Approximator and Embeddings
d31cf815
Fix auto pipeline + make style, quality
c85e46bd
keturn
commented
on 2025-06-12
make style
19733af2
![github-actions[bot]](https://avatars.githubusercontent.com/in/15368?v=4&s=60)
Apply style fixes
f49b149c
switch to new input ids
68b9cce8
Merge branch 'main' into chroma
ad01d636
fix # Copied from error
e97a4dd0
remove # Copied from on protected members
fd369246
try to fix import
2bc51c83
fix import
523150fb
make fix-copes
c330f08f
revert style fix
381e64b9
Merge remote-tracking branch '11698/chroma' into chroma-final
f35ec17a
update chroma transformer params
35dc65b7
update chroma transformer approximator init params
74fe45e8
update to pad tokens
926dcc63
fix batch inference
89faa71f
Make more pipeline tests work
829c6f19
Merge branch 'main' into chroma
272685c0
Make most transformer tests work
87664933
Merge remote-tracking branch 'origin/chroma' into chroma
60e41b78
fix docs
28dea06b
make style, make quality
bea8b0d8
sayakpaul
commented
on 2025-06-13
skip batch tests
00ebba97
fix test skipping
2b6722ec
sayakpaul
commented
on 2025-06-13
fix test skipping again
de9a07fc
DN6
commented
on 2025-06-13
fix for tests
67355077
DN6
commented
on 2025-06-13
Merge branch 'chroma-fork' into chroma-final
f1be3ebc
Fix all pipeline test
b85229e2
Merge branch 'chroma-fork' into chroma-final
bf56c953
update
292469d7
push local changes, fix docs
178c4ec9
add encoder test, remove pooled dim
16b6e339
default proj dim
06fb9957
fix tests
49a4c8bc
fix equal size list input
3fe4ad67
update
41751a3e
push local changes, fix docs
fd3e9445
add encoder test, remove pooled dim
8694f2ce
default proj dim
4e24f26d
fix tests
0978b609
fix equal size list input
c711e8f1
Revert "fix equal size list input"
589e939e
Merge branch 'chroma-fork' into chroma-final
2b559e9b
update
a967e66d
update
4f00bae5
update
0497faa3
update
e10f7015
update
d267bb69
DN6
approved these changes
on 2025-06-14
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub