Commit Graph

7 Commits

Author SHA1 Message Date
filipstrand
b5db14b3c6 Add descriptive comments for attention computations 2025-02-02 19:12:07 +01:00
filipstrand
9954e7fb45 Unify attention mechanism in single and joint attention blocks 2025-01-14 20:16:48 +01:00
filipstrand
f0e82ad67f Optimization: Use fast SDPA (no speedup for me on tests, but a bit cleaner)
Had to update reference test images, but visually no difference
2025-01-14 20:16:23 +01:00
filipstrand
bae49c8145 Dreambooth v1 2024-12-22 19:42:37 +01:00
filipstrand
6ef9ceb259 Fix linting issues 2024-09-21 19:59:12 +02:00
filipstrand
9e6ace4cc3 Apply ruff formating (minor manual modifications to some files which are subsequently excluded) 2024-09-21 19:59:12 +02:00
filipstrand
d2b38011ad Rename flux_1 to mflux 2024-09-06 21:09:16 +02:00