feat: add support for SD2.x with TINY U-Nets (#939)

This commit is contained in:
akleine
2025-11-09 15:47:37 +01:00
committed by GitHub
parent 0fa3e1a383
commit d2d3944f50
5 changed files with 68 additions and 47 deletions

View File

@@ -1788,6 +1788,9 @@ SDVersion ModelLoader::get_sd_version() {
if (is_inpaint) {
return VERSION_SD2_INPAINT;
}
if (!has_middle_block_1) {
return VERSION_SD2_TINY_UNET;
}
return VERSION_SD2;
}
return VERSION_COUNT;