Commit Graph

14 Commits

Author SHA1 Message Date
filipstrand
d2296ec197 Fix error in lora_layers causing lora training to fail 2025-06-12 23:38:52 +02:00
Filip Strand
847404748f
Generalise in-context functionality (#203)
Co-authored-by: claude[bot] <209825114+claude[bot]@users.noreply.github.com>
Co-authored-by: filipstrand <filipstrand@users.noreply.github.com>
2025-06-12 09:01:07 +02:00
Aarni Koskela
ae57884387 Fix type annotations for tuples 2025-04-25 15:30:47 +03:00
Aarni Koskela
ca2f96d75e Allow nullish zip_path for convenience 2025-04-25 14:00:47 +03:00
Aarni Koskela
83a7bf3519 Clean up fmt: off comments
* Use magic trailing commas instead of disabling formatting to keep args on separate lines
* Scope `fmt: off`s better where that's not possible
2025-03-31 00:55:38 +02:00
filipstrand
273431f010 Remove redundant is_mflux weight attribute 2025-03-15 11:45:47 +01:00
Anthony Wu
ce7dfeabfd remedy missed ruff check/formats in 0.6.0 release 2025-03-07 20:27:41 -08:00
Dana Case
96ba683181 clear prompt cache for training 2025-03-04 15:14:22 -08:00
filipstrand
3cecfae75b Refactor model config 2025-02-12 22:27:34 +01:00
filipstrand
0d9a5d626f Refactor: Transformer & Flux 2025-02-02 19:12:07 +01:00
filipstrand
f94edc4c3b Some fixes: Removed time in plotting, fix off-by-one error in training spec, update y-axis on loss plot 2025-01-27 23:48:03 +01:00
Alessandro
401da9fd75 Rebased on main (by filipstrand) 2025-01-27 23:15:27 +01:00
filipstrand
6fbbc9e5f7 Update missing model config lookup 2025-01-19 23:23:19 +01:00
filipstrand
bae49c8145 Dreambooth v1 2024-12-22 19:42:37 +01:00