✨ [Core] Add FreeU mechanism #5164
✨ Added Fourier filter function to upsample blocks
eec915a3
🔧 Update Fourier_filter for float16 support
8f2ee6b3
✨ Added UNetFreeUConfig to UNet model for FreeU adaptation 🛠️
ea33fc53
move unet to its original form and add fourier_filter to torch_utils.
6d77fff8
implement freeU enable mechanism
7c4676b4
implement disable mechanism
a152d910
resolution index.
c0de18df
correct resolution idx condition.
a0ced0ec
fix copies.
19d5ab01
no need to use resolution_idx in vae.
f3f94415
spell out the kwargs
266987c8
proper config property
83846f5e
fix attribution setting
cb6ebe8a
place unet hasattr properly.
9c8eacd6
fix: attribute access.
988ef763
proper disable
755a5c8e
remove validation method.
37b091e4
debug
4fd4adfd
debug
40291baa
debug
2e15e94e
debug
1e9c79ab
debug
883fc9bc
debug
ba08f301
potential fix.
540974aa
add: doc.
d38c251e
fix copies
785c0a07
add: tests.
466e0548
add: support freeU in SDXL.
80e560ed
set default value of resolution idx.
670b34b4
set default values for resolution_idx.
c5fc9383
fix copies
05ea56e3
fix rest.
93ee8672
fix copies
535eb590
address PR comments.
d277e643
sayakpaul
changed the title ✨ Added Fourier filter function to upsample blocks ✨ [Core] Add FreeU mechanism 2 years ago
run fix-copies
bb2d3683
move apply_free_u to utils and other minors.
e95b186a
introduce support for video (unet3D)
902cf7db
minor ups
15b1052f
consistent fix-copies.
7dcc939b
consistent stuff
3bf28bbe
Merge branch 'main' into add-freeU
64ade677
Merge branch 'main' into add-freeU
08d61fa0
fix-copies
d68663cb
add: rest
0e0af081
add: docs.
8d0a204d
fix: tests
55ad5355
fix: doc path
797b4b95
Merge branch 'main' into add-freeU
86419b91
Merge branch 'main' into add-freeU
5e27ff75
DN6
commented
on 2023-10-04
DN6
approved these changes
on 2023-10-04
Merge branch 'main' into add-freeU
2ad4953a
Apply suggestions from code review
1a8e5d16
style up
adc9d5c8
move to techniques.
c4f99d4b
add: slow test for sd freeu.
518e4b13
add: slow test for sd freeu.
dee3781e
add: slow test for sd freeu.
fc39d229
add: slow test for sd freeu.
053f3edd
add: slow test for sd freeu.
d8ef3a1f
add: slow test for sd freeu.
3da96e2f
add: slow test for video with freeu
d8c87712
add: slow test for video with freeu
0d34cf04
add: slow test for video with freeu
8e72f850
style
aa1a061e
sayakpaul
merged
84b82a6c
into main 2 years ago
kadirnar
deleted the add-freeU branch 2 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub