[Core] Introduce class variants for `Transformer2DModel` #7647
sayakpaul
merged
66 commits into
main
from
transormer2d-variants
init for patches
fa5a0bda
finish patched model.
05c03820
continuous transformer
e041313a
vectorized transformer2d.
ad0d71f9
style.
302bd022
inits.
ed2ddd11
sayakpaul
commented
on 2024-04-12
sayakpaul
commented
on 2024-04-12
sayakpaul
commented
on 2024-04-12
sayakpaul
changed the title
sayakpaul
changed the title
fix-copies.
9c83d68d
Merge branch 'main' into transormer2d-variants
a22ff0f3
merge main
c2da87bb
Merge branch 'main' into transormer2d-variants
89a09551
Merge branch 'main' into transormer2d-variants
20b06eee
introduce DiTTransformer2DModel.
47cc4741
sayakpaul
changed the title
fixes
fbade3c9
Merge branch 'main' into transormer2d-variants
a48453e8
sayakpaul
commented
on 2024-05-01
sayakpaul
commented
on 2024-05-01
sayakpaul
commented
on 2024-05-01
Merge branch 'transormer2d-variants' of https://github.com/huggingfac…
db5d0db4
DN6
commented
on 2024-05-01
Merge branch 'main' into transormer2d-variants
b54008f2
use REMAPPING as suggested by @DN6
4629163c
better logging.
d3cba344
add pixart transformer model.
94d14dc0
inits.
0043aa1d
caption_channels.
9db93e0b
attention masking.
82046e4a
sayakpaul
commented
on 2024-05-02
fix use_additional_conditions.
2c16cb90
remove print.
3a51d537
debug
ffed0b2c
flatten
c75a8c73
fix: assertion for sigma
08bf2289
sayakpaul
commented
on 2024-05-02
sayakpaul
changed the title
DN6
commented
on 2024-05-07
Merge branch 'main' into transormer2d-variants
c6471eb8
DN6
commented
on 2024-05-07
merge main and resolve conflicts.
a1dbf7c4
handle remapping for modeling_utils
15498f9b
add tests for dit transformer2d
d51a9f1e
quality
4576aa63
placeholder for pixart tests
64e3343a
sayakpaul
commented
on 2024-05-15
pixart tests
9d016fcd
add _no_split_modules
f4069a53
add docs.
57d4f14a
check
b707f004
check
0d28ea7d
check
158d6270
check
6b7f0d71
fix tests
6cdecf1a
fix tests
57f367c6
Merge branch 'main' into transormer2d-variants
2acaa99c
move Transformer output to modeling_output
95949680
move errors better and bring back use_additional_conditions attribute.
e0f6b08b
add unnecessary things from DiT.
ae753466
clean up pixart
57e31ec8
fix remapping
13217e80
Merge branch 'main' into transormer2d-variants
9354ccb0
fix device_map things in pixart2d.
d6d02bf0
Merge branch 'main' into transormer2d-variants
b4a1c08d
sayakpaul
commented
on 2024-05-17
Merge branch 'main' into transormer2d-variants
d1c6550e
replace Transformer2DModel with appropriate classes in dit, pixart tests
a6245ec7
Merge branch 'main' into transormer2d-variants
b50187e9
empty
5b3028eb
legacy mixin classes./
cb3bf767
DN6
commented
on 2024-05-21
DN6
commented
on 2024-05-21
DN6
commented
on 2024-05-21
DN6
commented
on 2024-05-21
use a remapping dict for fetching class names.
1dbdb80c
Merge branch 'main' into transormer2d-variants
3e8869d9
Merge branch 'main' into transormer2d-variants
58983db6
change to specifc model types in the pipeline implementations.
0b198acd
DN6
commented
on 2024-05-28
DN6
commented
on 2024-05-28
Merge branch 'main' into transormer2d-variants
a43a584a
move _fetch_remapped_cls_from_config to modeling_loading_utils.py
fa692c63
fix dependency problems.
b111afc9
DN6
commented
on 2024-05-30
DN6
approved these changes
on 2024-05-30
add deprecation note.
8edcf0b5
Merge branch 'main' into transormer2d-variants
cb0f9943
Merge branch 'main' into transormer2d-variants
3406d344
yiyixuxu
commented
on 2024-06-08
Login to write a write a comment.
Login via GitHub