upstream-archive/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-07-30 23:00:41 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: add support for Flux Controls and Flex.2 (#692)

Browse Source
This commit is contained in:
stduhpf
2025-10-10 18:06:57 +02:00
committed by GitHub
parent 35843c77ea
commit 11f436c483
7 changed files with 156 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
model.cpp
View File

@@ -1803,10 +1803,15 @@ SDVersion ModelLoader::get_sd_version() {
}
if (is_flux) {
is_inpaint = input_block_weight.ne[0] == 384;
if (is_inpaint) {
if (input_block_weight.ne[0] == 384) {
return VERSION_FLUX_FILL;
}
if (input_block_weight.ne[0] == 128) {
return VERSION_FLUX_CONTROLS;
}
if (input_block_weight.ne[0] == 196) {
return VERSION_FLEX_2;
}
return VERSION_FLUX;
}