Commit Graph

249 Commits

Author SHA1 Message Date
Anthony Wu
d20f2833da
🪫 Battery Saver - manage power over long batch operations (#189)
Co-authored-by: Anthony Wu <pls-file-gh-issue@users.noreply.github.com>
Co-authored-by: filipstrand <strand.filip@gmail.com>
2025-05-19 17:13:41 +02:00
Anthony Wu
4b56633e6c
Add support for --prompt-file for dynamic prompt updates in large batches (#185)
Co-authored-by: Anthony Wu <pls-file-gh-issue@users.noreply.github.com>
Co-authored-by: filipstrand <strand.filip@gmail.com>
2025-05-19 17:10:52 +02:00
Alessandro
3b47c0b7e5
Possibility to balance the Redux function (#190)
Co-authored-by: Alessandro Rizzo <alessandrorizzo@Alessandros-Mac-Studio.local>
Co-authored-by: filipstrand <strand.filip@gmail.com>
2025-05-19 15:28:15 +02:00
filipstrand
42b4fd6d90 Refactor DepthPro object 2025-05-10 13:40:21 +02:00
filipstrand
4cb8a5b203 Better init pattern for depth model 2025-05-10 13:40:21 +02:00
filipstrand
5f0f39f362 Update calls in MemorySaver 2025-05-07 22:00:13 +02:00
filipstrand
393a60fb2b Small fix for LoRA scales 2025-05-06 13:22:42 +02:00
Anthony Wu
136b76de01 propose --save-full-image option 2025-05-01 00:34:49 -07:00
Anthony Wu
8b828ced27 In-Context generation: allow lora composition 2025-04-30 22:00:33 -07:00
Filip Strand
3af1cc18df
Merge pull request #175 from akx/tup-typ
Fix type annotations for tuples
2025-04-25 14:45:04 +02:00
Aarni Koskela
da5477bdbf Apply no-implicit-optional fixes 2025-04-25 15:32:27 +03:00
Aarni Koskela
ae57884387 Fix type annotations for tuples 2025-04-25 15:30:47 +03:00
Aarni Koskela
79ca146653 Fix type for Img2Img.image_path 2025-04-25 14:03:29 +03:00
Aarni Koskela
ca2f96d75e Allow nullish zip_path for convenience 2025-04-25 14:00:47 +03:00
Aarni Koskela
dc175f3d14 Fix str/Path related typing 2025-04-25 13:59:42 +03:00
Aarni Koskela
effff3273a Normalize Path import style 2025-04-25 13:50:00 +03:00
filipstrand
7cc4099995 Fix command line args 2025-04-25 10:44:13 +02:00
filipstrand
74ab9acf15 Remove torchvision dependency 2025-04-24 15:43:38 +02:00
filipstrand
d376f9c931 FLUX.1 Tools | Redux & Depth 2025-04-24 09:08:21 +02: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
Aarni Koskela
7a9d9e667b Remove duplicate mask_util file
Looks like an inadvertent copy of mflux/flux_tools/fill/mask_util.py
2025-03-30 20:09:58 +03:00
Serkan Sakar
d225d1bbb9 More low-ram improvements 2025-03-22 10:46:21 +01:00
filipstrand
d4d98186e2 FLUX.1 Tools | Fill 2025-03-16 11:12:34 +01:00
Filip Strand
195ed134a0
Merge pull request #141 from anthonywu/prepare-outpaint-scaffolding
Image Outpaint Feature: prepare the CLI and ImageUtil helpers for extended canvas
2025-03-15 12:14:07 +01:00
filipstrand
273431f010 Remove redundant is_mflux weight attribute 2025-03-15 11:45:47 +01:00
filipstrand
696762af2e Save mflux versions in weight metadata 2025-03-15 11:45:47 +01:00
Anthony Wu
8eeecde8f7
Merge branch 'main' into prepare-outpaint-scaffolding 2025-03-13 21:38:42 -07:00
Goekdeniz-Guelmez
f5f8d45eaa initial commmit 2025-03-13 09:40:35 +01:00
Anthony Wu
13121684ac Image Outpaint Feature: prepare the CLI and ImageUtil helpers 2025-03-09 18:38:44 -07:00
Anthony Wu
04d24234f1 Fix: image gen should stop even if no callbacks are registered 2025-03-09 07:42:12 -07: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
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
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
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
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
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
filipstrand
3cecfae75b Refactor model config 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
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
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
fa99cf0cc8 Refactor: Controlnet transformer & Flux controlnet 2025-02-02 16:45:27 +01:00
filipstrand
010610e33b Remove comments 2025-01-28 00:04:35 +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
Filip Strand
cb93e1aaf6
Merge pull request #120 from anthonywu/renamer-uv-script-tool
Batch Image Renamer tool as an isolated uv run script
2025-01-27 22:23:18 +01:00
Anthony Wu
9053bcd6d8 Batch Image Renamer tool as an isolated uv run script 2025-01-22 13:25:59 -08:00
Filip Strand
4bea7b9668
Merge pull request #118 from anthonywu/run-n-seeds-auto-seeds
support multiple seeds batch generation
2025-01-22 19:58:51 +01:00
filipstrand
807b243e56 Support variable numbers of transformer blocks 2025-01-22 08:18:39 +01:00
Anthony Wu
af01fab7d5 support multiple seeds batch generation 2025-01-21 18:14:24 -08:00
filipstrand
6fbbc9e5f7 Update missing model config lookup 2025-01-19 23:23:19 +01:00
filipstrand
398cf082c3 Fix string formating 2025-01-19 19:49:57 +01:00
filipstrand
21a98b1369 Small updates 2025-01-19 15:19:17 +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
4c0aff9168 Fix bug with quantization level being str instead of int 2024-12-23 12:21:06 +01:00
filipstrand
bae49c8145 Dreambooth v1 2024-12-22 19:42:37 +01:00
filipstrand
ce387921c5 Add named args for height and width for clarity 2024-10-29 20:12:50 +01:00
filipstrand
82973f9e95 Fix bug: swap width <--> height 2024-10-29 20:12:50 +01:00
filipstrand
4e2a10b1b9 Rewrite img2img latent creation to emphasize the linear interp between latent and noise 2024-10-29 19:19:59 +01:00
filipstrand
ea6202b49f Add img2img example in readme 2024-10-28 00:32:57 +01:00
Anthony Wu
960447f92e update type hint 2024-10-26 20:24:28 -07:00
Anthony Wu
98d7ef5751 update: do not require model arg, add controlnet_save_canny, fix bugs, add tests 2024-10-26 20:22:32 -07:00
Anthony Wu
ffaf562a8c fix type hint in test file 2024-10-26 10:29:33 -07:00
Anthony Wu
351cc8bbf1 support metadata files as CLI arg supplier 2024-10-20 09:18:49 -07:00
filipstrand
2ed0628e58 Light refactoring: Move latent creation to separate class 2024-10-20 11:49:43 +02:00
filipstrand
f08c667e1a Remove vae type alias from runtime_config.py 2024-10-20 11:05:45 +02:00
Anthony Wu
1160d4e8b2 controlnet files: revert the refactoring done in #77 2024-10-20 01:06:02 -07:00
Anthony Wu
6201a93d4a tests and fixes for image to image 2024-10-20 00:48:45 -07:00
Anthony Wu
f963b7a9b2 image to image implementation 2024-10-12 10:46:20 -07:00
filipstrand
9c7e857082 Add type hinting to CommandLineParser 2024-10-12 14:35:11 +02:00
Anthony Wu
796e9e8305 refactor argparser 2024-10-11 15:03:52 -07:00
Anthony Wu
9b6cd01171 introduce ruff isort 2024-10-10 11:56:14 -07:00
Filip Strand
faaf905667
Merge pull request #71 from anthonywu/typos-check
introduce pre-commit typos check
2024-10-10 20:31:21 +02:00
filipstrand
8ec5410a56 Mark stepwise feature as experimental for now 2024-10-10 07:43:19 +02:00
Anthony Wu
19867d5bb7 introduce pre-commit typos check 2024-10-09 14:20:49 -07:00
filipstrand
e908c1cede Add stepwise handler 2024-10-09 22:29:36 +02:00
Anthony Wu
5de208d1ba PR review 2024-10-06 14:59:15 -10:00
Anthony Wu
fcc9b21d19 option to save stepwise and composite images, add interruptibility 2024-09-23 15:20:25 -07:00
filipstrand
a942ca8d67 Rename controlnet path arg 2024-09-23 20:22:47 +02:00
filipstrand
ec8b42e521 Rename save arg 2024-09-23 20:22:30 +02:00
filipstrand
ebff19b48a Use None instead of empty string for image metadata 2024-09-23 20:13:35 +02:00
filipstrand
dfdd543227 Add controlnet example 2024-09-23 20:13:35 +02:00
filipstrand
91b0c466fb Avoid circular dep. issue (maybe better way to do this) 2024-09-23 20:13:35 +02:00
filipstrand
15ecaa7370 Rename util method 'save' to 'save_image' 2024-09-23 20:13:35 +02:00
filipstrand
3af6bcbfe1 Add ability to save the canny controlnet image early on in the generation
(useful for inspecting if edge detection is as expected)
2024-09-23 20:13:35 +02:00