Commit Graph

40 Commits

Author SHA1 Message Date
Serkan Sakar
f65286e1fb add memory saver to controlnet 2025-03-02 19:28:19 +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
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
fa99cf0cc8 Refactor: Controlnet transformer & Flux controlnet 2025-02-02 16:45:27 +01:00
filipstrand
807b243e56 Support variable numbers of transformer blocks 2025-01-22 08:18:39 +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
2ed0628e58 Light refactoring: Move latent creation to separate class 2024-10-20 11:49:43 +02:00
Anthony Wu
1160d4e8b2 controlnet files: revert the refactoring done in #77 2024-10-20 01:06:02 -07:00
Anthony Wu
f963b7a9b2 image to image implementation 2024-10-12 10:46:20 -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
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
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
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
d799a924c4 Light refactor of controlnet class 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
Anthony Wu
8b47492f1b more qol suggestions 2024-09-21 15:03:35 -07: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
abd7893612 Fix hidden_states in single_transformer_blocks for controlnet transformer 2024-09-21 14:38:33 +02:00
filipstrand
38c747b680 Extract model saving logic to separate class 2024-09-17 07:35:42 +02:00
filipstrand
a30dc4d5cc Rename image.py to generated_image.py 2024-09-17 07:35:42 +02:00
filipstrand
bce18747ff Fix formatting 2024-09-17 07:35:42 +02:00
filipstrand
fc8099e908 Move new controlnet module to its own file 2024-09-17 07:35:42 +02:00
filipstrand
ad6f4ca8af Create util class for controlnet 2024-09-17 07:35:42 +02:00
filipstrand
78bbf3d8fb Fix formatting and some warnings 2024-09-17 07:35:41 +02:00
filipstrand
25e375d45d Small additions and refactoring 2024-09-17 06:35:06 +02:00
Fabio
a2900c3b13 update generated image annotations 2024-09-15 21:54:17 +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