Commit Graph

140 Commits

Author SHA1 Message Date
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
filipstrand
7b4aab9c1b Move actual image saving logic to ImageUtil 2024-09-23 20:13:35 +02:00
filipstrand
d799a924c4 Light refactor of controlnet class 2024-09-23 20:13:35 +02:00
filipstrand
5737b27164 Save controlnet_image and mflux version in metadata 2024-09-23 20:13:35 +02:00
filipstrand
cda698df8d Move controlnet specific weight handling to separate class 2024-09-23 20:13:35 +02:00
filipstrand
8678102321 Fix controlnet input arg name and group controlnet arguments 2024-09-23 19:18:11 +02:00
filipstrand
56f0c8361d Add TOC and some emojis for the README and new controlnet section 2024-09-23 19:18:11 +02:00
filipstrand
3ac54a9f42 Add ability to save model with applied LoRA 2024-09-22 11:20:31 +02: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