Commit Graph

96 Commits

Author SHA1 Message Date
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
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
c170e89977 remove unnecessary imports 2024-09-21 10:39:01 +02:00
filipstrand
5a7157e7e8 Simplify import paths for controlnet file 2024-09-21 10:34:35 +02:00
filipstrand
e007402cfa Merge remote-tracking branch 'refs/remotes/mflux/main' into easy-import-dx 2024-09-21 10:33:36 +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
Anthony Wu
43618154d3 remove apparently unnecessary import of the src/mflux directory 2024-09-16 15:04:30 -07:00
Anthony Wu
0812d6252c simplify import paths of frequently used items in scripts and demos 2024-09-16 10:27:04 -07: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
a04d139ff5 further cleanup 2024-09-14 20:04:08 +02:00
Fabio
2a2a3fff63 delete egg-info 2024-09-14 20:01:33 +02:00
Fabio
12a6386fca Merge branch 'main' into feature/add_controlnet 2024-09-14 19:58:51 +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
filipstrand
db3e495be6 Remove restriction on lora weights 2024-09-13 21:50:48 +02:00
filipstrand
d10e6f4019 Add support for converting between LoRA formats 2024-09-13 20:37:56 +02:00
Fabio
1b9c645503 WIP controlnet 2024-09-13 00:28:43 +02:00
filipstrand
7d27f1c4c1 Adjust FF block only if it exists 2024-09-12 21:18:01 +02:00
filipstrand
1567a01745 Rename run to generate 2024-09-06 22:27:24 +02:00
filipstrand
c0d419777e Add quantize option to from_alias initializer 2024-09-06 22:11:00 +02:00
filipstrand
af73966f11 Fix system paths 2024-09-06 21:45:02 +02:00
filipstrand
b9ff93ea3f Fix imports 2024-09-06 21:15:46 +02:00
filipstrand
23e44c768b Rename main.py to run.py 2024-09-06 21:09:50 +02:00
filipstrand
b8102f2a2f Move main.py and save.py one level down 2024-09-06 21:09:44 +02:00
filipstrand
915f3c2950 Move flux.py to separate package 2024-09-06 21:09:26 +02:00
filipstrand
d2b38011ad Rename flux_1 to mflux 2024-09-06 21:09:16 +02:00
filipstrand
6fb51170de Also save LoRA metadata 2024-09-06 20:09:27 +02:00
filipstrand
6d05f5b27e Fix saving bug 2024-09-06 20:09:17 +02:00
filipstrand
fab10cc046 Add proper image class and save metadata 2024-09-06 19:38:34 +02:00
Filip Strand
97dc64436e
Merge pull request #31 from Xuzzo/fix/width_height_mixup
Fix width and height in config
2024-09-06 10:02:46 +02:00
filipstrand
1d8714d3b1 Add some example runs 2024-09-05 23:21:48 +02:00
filipstrand
a0ee9a7a8a Rename apply-lora arg 2024-09-05 21:50:31 +02:00
Fabio
0a2f21af5a fix widht <-> height bug 2024-09-05 20:06:59 +02:00
filipstrand
96848e320f Add support for multiple LoRAs again 2024-09-05 19:54:46 +02:00
filipstrand
ff9f998638 Small fixes 2024-09-04 23:16:02 +02:00
Filip Strand
d5f5f9607e
Fix type
Co-authored-by: Fabio Peruzzo <39661245+Xuzzo@users.noreply.github.com>
2024-09-04 23:14:38 +02:00
filipstrand
f28fffe0cf Point out the .safetensors file again 2024-09-04 22:44:11 +02:00