Commit Graph

325 Commits

Author SHA1 Message Date
filipstrand
cb09eacb69 Update readme with new 4-bit download location 2025-03-15 10:00:47 +01:00
filipstrand
4f88376830 v.0.6.2 2025-03-13 22:11:16 +01:00
Filip Strand
0c963b1e40
Merge pull request #147 from Goekdeniz-Guelmez/fix-parser-in-save
Fix the parser error in save
2025-03-13 22:06:39 +01:00
Goekdeniz-Guelmez
f5f8d45eaa initial commmit 2025-03-13 09:40:35 +01:00
Filip Strand
0e9c6212de
Merge pull request #142 from anthonywu/quick-fix-auto-seeds-test-case
Fix test: auto seed count == 1 sporadic test failure
2025-03-10 07:39:24 +01:00
Filip Strand
a1325ccba3
Merge pull request #140 from anthonywu/restore-keyboard-interrupt
Fix: image gen should stop even if no callbacks are registered
2025-03-10 07:38:36 +01:00
Anthony Wu
4ad31cab6b fix auto seed count == 1 sporadic test failure 2025-03-09 18:45:05 -07:00
Anthony Wu
04d24234f1 Fix: image gen should stop even if no callbacks are registered 2025-03-09 07:42:12 -07:00
Filip Strand
9f1c6dd408
Merge pull request #138 from anthonywu/cleanup-0.6.0
remedy missed ruff check/formats in 0.6.0 release
2025-03-08 11:01:55 +01:00
Anthony Wu
ce7dfeabfd remedy missed ruff check/formats in 0.6.0 release 2025-03-07 20:27:41 -08:00
filipstrand
e13181e176 Add after_loop callback to in context generation 2025-03-05 08:32:26 +01:00
filipstrand
368fa0faf7 Upload IC-LoRA test image 2025-03-05 08:19:39 +01:00
filipstrand
c5991d54d4 Upload IC-LoRA reference test image 2025-03-05 08:17:00 +01:00
Filip Strand
fb15edc7fc
Merge pull request #127 from filipstrand/in-context-lora
In-Context LoRA
2025-03-05 08:02:33 +01:00
filipstrand
ab3d7abdd2 Update readme to be up-to-date with the newest features 2025-03-05 08:00:01 +01:00
filipstrand
f75593b1dc New: Add In-Context-LoRA feature 2025-03-05 07:55:21 +01:00
filipstrand
c8c35ac8c9 Add ability to download LoRAs automatically from Huggingface 2025-03-05 07:55:21 +01:00
filipstrand
5ec7ab5b41 Renaming: Remove 'init' prefix for a more general interface 2025-03-05 07:30:46 +01:00
Filip Strand
552eb6a3bb
Merge pull request #136 from DanaCase/clear_prompt_cache
clear prompt cache for training
2025-03-05 07:30:16 +01:00
Filip Strand
ceea386361
Merge pull request #135 from DanaCase/test_low_ram
Update arg parser test fixture
2025-03-05 07:23:51 +01:00
Dana Case
96ba683181 clear prompt cache for training 2025-03-04 15:14:22 -08:00
Dana Case
50734795b9 add general arguments to train as well 2025-03-03 19:21:01 -08:00
Dana Case
bf1e366e43 formatting 2025-03-03 18:54:49 -08:00
Filip Strand
c552efe659
Merge pull request #131 from ssakar/memory_saver
Memory usage optimizations
2025-03-02 22:52:57 +01:00
Serkan Sakar
6006806e6d
Merge branch 'filipstrand:main' into memory_saver 2025-03-02 19:38:53 +01:00
Serkan Sakar
f65286e1fb add memory saver to controlnet 2025-03-02 19:28:19 +01:00
Filip Strand
d1a5f55378
Merge pull request #133 from filipstrand/pre-test-cleanup
Pre test cleanup
2025-03-01 22:26:26 +01:00
filipstrand
06f336c714 Pre test cleanup 2025-03-01 22:10:10 +01:00
Filip Strand
d3ce9b67ec
Merge pull request #132 from filipstrand/add-option-to-override-saved-images
Add option to override saved images
2025-03-01 21:47:49 +01:00
filipstrand
09430350f1 Override previously saved test images 2025-03-01 21:37:03 +01:00
filipstrand
d80b6e520f Add option to override saved images 2025-03-01 21:34:26 +01:00
Serkan Sakar
50752f58f4 Memory usage optimizations 2025-02-28 20:43:01 +01:00
filipstrand
8e5e09f7d8 Allow latest mlx 2025-02-19 08:09:34 +01:00
filipstrand
b733452878 Temporarily cap mlx at <0.23.0 until tests pass 2025-02-17 20:42:43 +01:00
Filip Strand
dca6c0acea
Merge pull request #126 from filipstrand/img2img-special-case
Img2img special case fix
2025-02-14 10:41:02 +01:00
filipstrand
821238284a Fix: Handle edge case for img2img 2025-02-14 05:21:35 +01:00
filipstrand
74173f543e Let before-loop callback accept latents and let StepwiseHandler include initial latent 2025-02-14 05:21:18 +01:00
Filip Strand
0417852a9b
Merge pull request #125 from filipstrand/more-quantization
More quantization options
2025-02-12 23:05:08 +01:00
filipstrand
3cecfae75b Refactor model config 2025-02-12 22:27:34 +01:00
filipstrand
f21710eb7e Update reference test images (minor changes due to updated quantization call) 2025-02-12 22:27:34 +01:00
filipstrand
4563bdd34b Update licence 2025-02-12 22:27:34 +01:00
filipstrand
a26277a7dd Require mlx >= v0.22.0 2025-02-12 22:27:34 +01:00
filipstrand
e3b2a65aa6 Allow 3-bit and 6-bit quantization (mlx > v0.21.0) 2025-02-12 22:27:34 +01:00
filipstrand
09d75e46b8 Add PromptCache: Small optimization for repeated generations with same prompt 2025-02-12 22:27:34 +01:00
Filip Strand
2d481e77b2
Merge pull request #124 from filipstrand/refactor-and-add-callback-mechanism
Refactor and add callback mechanism
2025-02-10 18:53:52 +01:00
filipstrand
3601baf75e Implement general callback mechanism 2025-02-09 15:27:06 +01:00
filipstrand
e4af40e4ee Refactor: Remove ConfigControlnet 2025-02-09 14:37:23 +01:00
filipstrand
22f98fc5a4 Refactor: Add separate Flux initializer 2025-02-09 14:37:23 +01:00
Filip Strand
8f82549f43
Merge pull request #122 from filipstrand/transformer-refactor
Transformer refactor
2025-02-02 19:13:45 +01:00
filipstrand
b5db14b3c6 Add descriptive comments for attention computations 2025-02-02 19:12:07 +01:00