Commit Graph

22 Commits

Author SHA1 Message Date
filipstrand
09d75e46b8 Add PromptCache: Small optimization for repeated generations with same prompt 2025-02-12 22:27:34 +01:00
filipstrand
b5db14b3c6 Add descriptive comments for attention computations 2025-02-02 19:12:07 +01:00
filipstrand
0485cf07c2 Refactor: Transformer blocks 2025-02-02 19:12:07 +01:00
filipstrand
0d9a5d626f Refactor: Transformer & Flux 2025-02-02 19:12:07 +01:00
filipstrand
807b243e56 Support variable numbers of transformer blocks 2025-01-22 08:18:39 +01:00
Anthony Wu
344ac90c74 comprehensive ModelConfig refactor to support compatible HuggingFace dev/schnell models 2025-01-17 06:27:33 -08:00
filipstrand
16a0225b41 Specify precision for causal attention mask 2025-01-14 21:32:02 +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
Anthony Wu
9b6cd01171 introduce ruff isort 2024-10-10 11:56:14 -07:00
Anthony Wu
19867d5bb7 introduce pre-commit typos check 2024-10-09 14:20:49 -07:00
Anthony Wu
8b47492f1b more qol suggestions 2024-09-21 15:03:35 -07: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
38c747b680 Extract model saving logic to separate class 2024-09-17 07:35:42 +02:00
filipstrand
78bbf3d8fb Fix formatting and some warnings 2024-09-17 07:35:41 +02:00
Fabio
dee452b3bb fixes from review 2024-09-15 21:37:29 +02:00
Fabio
369bfc0b53 add cli, controlnet config and tidy up 2024-09-14 19:58:12 +02:00
Fabio
3dba1e38e1 controlnet - first working version 2024-09-14 14:37:30 +02:00
Fabio
1b9c645503 WIP controlnet 2024-09-13 00:28:43 +02:00
filipstrand
d2b38011ad Rename flux_1 to mflux 2024-09-06 21:09:16 +02:00