Commit Graph

18 Commits

Author SHA1 Message Date
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