diffusers
[DC-AE] Add the official Deep Compression Autoencoder code(32x,64x,128x compression ratio);
#9708
Merged
Overview
Commits
101
Changes
View On GitHub

[DC-AE] Add the official Deep Compression Autoencoder code(32x,64x,128x compression ratio); #9708

a-r-r-o-w merged 101 commits into huggingface:main from lawrence-cj:DC-AE
lawrence-cj
lawrence-cj first add a script for DC-AE;
6e616a9c
sayakpaul sayakpaul marked this pull request as draft 341 days ago
Abhinay1997
chenjy2003 Merge remote-tracking branch 'upstream/main' into DC-AE
d2e187a0
chenjy2003 DC-AE init
90e89393
chenjy2003 replace triton with custom implementation
825c975c
lawrence-cj 1. rename file and remove un-used codes;
3a44fa4c
sayakpaul
sayakpaul commented on 2024-10-23
chenjy2003 no longer rely on omegaconf and dataclass
55b2615d
chenjy2003 merge
6fb7fdb8
chenjy2003 Merge remote-tracking branch 'upstream/main' into DC-AE
c323e76b
chenjy2003 replace custom activation with diffuers activation
da7caa58
chenjy2003 remove dc_ae attention in attention_processor.py
fb6d92a4
chenjy2003 iinherit from ModelMixin
5e63a1ae
chenjy2003 inherit from ConfigMixin
72cce2b4
chenjy2003 dc-ae reduce to one file
8f9b4e43
chenjy2003 Merge remote-tracking branch 'upstream/main' into DC-AE
b7f68f9a
sayakpaul
sayakpaul commented on 2024-10-31
lawrence-cj Merge branch 'huggingface:main' into DC-AE
6d96b954
lawrence-cj Merge remote-tracking branch 'refs/remotes/origin/main' into DC-AE
3c3cc517
chenjy2003 update downsample and upsample
14486810
chenjy2003 merge
bf40fe85
chenjy2003 clean code
dd7718a2
chenjy2003 support DecoderOutput
19986a52
lawrence-cj Merge branch 'main' into DC-AE
3481e23c
sayakpaul
sayakpaul commented on 2024-11-09
yiyixuxu
yiyixuxu commented on 2024-11-12
lawrence-cj Merge branch 'main' into DC-AE
0e818df8
chenjy2003 remove get_same_padding and val2tuple
c6eb2336
chenjy2003 remove autocast and some assert
59de0a34
chenjy2003 update ResBlock
ea604a4d
chenjy2003 remove contents within super().__init__
80dce027
lawrence-cj lawrence-cj requested a review from sayakpaul sayakpaul 312 days ago
lawrence-cj lawrence-cj requested a review from yiyixuxu yiyixuxu 312 days ago
lawrence-cj Update src/diffusers/models/autoencoders/dc_ae.py
1752afdd
chenjy2003 remove opsequential
883bcf4f
chenjy2003 Merge branch 'DC-AE' of github.com:lawrence-cj/diffusers into DC-AE
25ae3890
chenjy2003 update other blocks to support the removal of build_norm
96e844b7
sayakpaul Merge branch 'main' into DC-AE
59b6e257
chenjy2003 remove build encoder/decoder project in/out
7ce9ff2f
chenjy2003 Merge branch 'DC-AE' of github.com:lawrence-cj/diffusers into DC-AE
30d6308d
chenjy2003 remove inheritance of RMSNorm2d from LayerNorm
cab56b16
yiyixuxu
yiyixuxu commented on 2024-11-20
yiyixuxu yiyixuxu requested a review from a-r-r-o-w a-r-r-o-w 308 days ago
chenjy2003 remove reset_parameters for RMSNorm2d
b42bb548
chenjy2003 remove device and dtype in RMSNorm2d __init__
2e04a996
lawrence-cj Update src/diffusers/models/autoencoders/dc_ae.py
b4f75f22
lawrence-cj Update src/diffusers/models/autoencoders/dc_ae.py
c82f828b
lawrence-cj Update src/diffusers/models/autoencoders/dc_ae.py
22ea5fd7
chenjy2003 remove op_list & build_block
4f5cbb43
chenjy2003 remove build_stage_main
2f6bbad8
lawrence-cj Merge branch 'main' into DC-AE
44957833
a-r-r-o-w
a-r-r-o-w commented on 2024-11-27
lawrence-cj
chenjy2003
chenjy2003 change file name to autoencoder_dc
4d3c0266
chenjy2003 Merge branch 'DC-AE' of github.com:lawrence-cj/diffusers into DC-AE
e007057d
chenjy2003 move LiteMLA to attention.py
d3d9c840
a-r-r-o-w
chenjy2003
a-r-r-o-w
a-r-r-o-w
chenjy2003
lawrence-cj
chenjy2003
lawrence-cj align with other vae decode output;
be9826c5
lawrence-cj add DC-AE into init files;
20da2012
a-r-r-o-w update
5ed50e98
lawrence-cj
lawrence-cj make quality && make style;
2d59056c
a-r-r-o-w quick push before dgx disappears again
c1c02a20
a-r-r-o-w update
1f8a3b3b
a-r-r-o-w make style
7b9d7e55
a-r-r-o-w update
bf6c211e
a-r-r-o-w
a-r-r-o-w update
a2ec5f81
a-r-r-o-w fix
f5876c5b
a-r-r-o-w refactor
44034a6f
a-r-r-o-w refactor
63792412
a-r-r-o-w refactor
77571a8f
a-r-r-o-w update
c4d0867c
a-r-r-o-w possibly change to nn.Linear
0bdb7eff
a-r-r-o-w refactor
54e933b3
a-r-r-o-w Merge branch 'main' into aryan-dcae
babc9f59
a-r-r-o-w make fix-copies
3d5faaff
a-r-r-o-w resolve conflicts & merge
65edfa5e
a-r-r-o-w
HuggingFaceDocBuilderDev
a-r-r-o-w a-r-r-o-w requested a review from yiyixuxu yiyixuxu 297 days ago
chenjy2003
a-r-r-o-w
yiyixuxu
yiyixuxu commented on 2024-11-27
chenjy2003 replace vae with ae
ca3ac4de
chenjy2003 replace get_block_from_block_type to get_block
9ef7b597
chenjy2003 replace downsample_block_type from Conv to conv for consistency
074817cf
chenjy2003 add scaling factors
64de66a4
yiyixuxu yiyixuxu added roadmap
a-r-r-o-w incorporate changes for all checkpoints
0bda5c51
a-r-r-o-w
a-r-r-o-w make style
eb64d52c
a-r-r-o-w Merge branch 'main' into DC-AE
4a224ce9
lawrence-cj
a-r-r-o-w
a-r-r-o-w move mla to attention processor file; split qkv conv to linears
30c32383
a-r-r-o-w refactor
39a947ce
chenjy2003
a-r-r-o-w Merge branch 'main' into DC-AE
68f817ac
a-r-r-o-w add tests
da834d51
a-r-r-o-w
a-r-r-o-w
lawrence-cj
lawrence-cj lawrence-cj marked this pull request as ready for review 294 days ago
lawrence-cj Merge branch 'main' into DC-AE
632ad3ba
lawrence-cj
a-r-r-o-w
lawrence-cj
a-r-r-o-w from original file loader
d6c748c7
a-r-r-o-w Merge branch 'main' into DC-AE
46eb5043
a-r-r-o-w
a-r-r-o-w a-r-r-o-w requested a review from DN6 DN6 294 days ago
a-r-r-o-w add docs
31f9fc6c
a-r-r-o-w add standard autoencoder methods
6f29e2ab
yiyixuxu
yiyixuxu approved these changes on 2024-12-04
lawrence-cj
yiyixuxu
chenjy2003
a-r-r-o-w
a-r-r-o-w approved these changes on 2024-12-05
yiyixuxu combine attention processor
b6e8fbac
a-r-r-o-w fix tests
f862bae8
a-r-r-o-w update
f9fce24a
a-r-r-o-w Merge branch 'main' into DC-AE
e5947451
chenjy2003 minor fix
3c0b1ca3
chenjy2003 minor fix
91057d40
lawrence-cj Merge branch 'main' into DC-AE
67aa715e
chenjy2003 minor fix & in/out shortcut rename
eda66e19
chenjy2003 minor fix
e3d33e6a
chenjy2003
a-r-r-o-w Merge branch 'main' into DC-AE
cc97502c
a-r-r-o-w make style
2b370df4
a-r-r-o-w
chenjy2003
chenjy2003 fix paper link
94355abe
a-r-r-o-w Merge branch 'main' into DC-AE
a191f070
a-r-r-o-w update docs
116c0495
DN6
DN6 commented on 2024-12-06
DN6
DN6 commented on 2024-12-06
a-r-r-o-w update single file loading
b6e0aba6
a-r-r-o-w Merge branch 'main' into DC-AE
ec4e84fd
a-r-r-o-w make style
dbae8f1d
stevhliu
stevhliu approved these changes on 2024-12-06
a-r-r-o-w remove single file loading support; todo for DN6
042c2a0c
a-r-r-o-w Apply suggestions from code review
f2525b94
a-r-r-o-w Merge branch 'main' into DC-AE
d3d224c0
a-r-r-o-w
a-r-r-o-w add abstract
6122b84a
a-r-r-o-w
a-r-r-o-w approved these changes on 2024-12-06
a-r-r-o-w a-r-r-o-w merged cd892041 into main 292 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
roadmap
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum