Support sdxl ip2p

Author: stduhpf

model.cpp

@@ -1539,11 +1539,14 @@ SDVersion ModelLoader::get_sd_version() {
         }
     }
     bool is_inpaint = input_block_weight.ne[2] == 9;
-    bool is_ip2p =  input_block_weight.ne[2] == 8;
+    bool is_ip2p    = input_block_weight.ne[2] == 8;
     if (is_xl) {
         if (is_inpaint) {
             return VERSION_SDXL_INPAINT;
         }
+        if (is_ip2p) {
+            return VERSION_SDXL_PIX2PIX;
+        }
         return VERSION_SDXL;
     }
 
@@ -1559,7 +1562,7 @@ SDVersion ModelLoader::get_sd_version() {
         if (is_inpaint) {
             return VERSION_SD1_INPAINT;
         }
-        if(is_ip2p) {
+        if (is_ip2p) {
             return VERSION_SD1_PIX2PIX;
         }
         return VERSION_SD1;

model.h

@@ -26,6 +26,7 @@ enum SDVersion {
     VERSION_SD2_INPAINT,
     VERSION_SDXL,
     VERSION_SDXL_INPAINT,
+    VERSION_SDXL_PIX2PIX,
     VERSION_SVD,
     VERSION_SD3,
     VERSION_FLUX,
@@ -62,7 +63,7 @@ static inline bool sd_version_is_sd2(SDVersion version) {
 }
 
 static inline bool sd_version_is_sdxl(SDVersion version) {
-    if (version == VERSION_SDXL || version == VERSION_SDXL_INPAINT) {
+    if (version == VERSION_SDXL || version == VERSION_SDXL_INPAINT || version == VERSION_SDXL_PIX2PIX) {
         return true;
     }
     return false;
@@ -83,7 +84,7 @@ static inline bool sd_version_is_dit(SDVersion version) {
 }
 
 static inline bool sd_version_is_edit(SDVersion version) {
-    return version == VERSION_SD1_PIX2PIX;
+    return version == VERSION_SD1_PIX2PIX || version == VERSION_SDXL_PIX2PIX;
 }
 
 static bool sd_version_use_concat(SDVersion version) {

stable-diffusion.cpp

@@ -32,6 +32,7 @@ const char* model_version_to_str[] = {
     "SD 2.x Inpaint",
     "SDXL",
     "SDXL Inpaint",
+    "SDXL Instruct-Pix2Pix",
     "SVD",
     "SD3.x",
     "Flux",

unet.hpp

@@ -207,7 +207,7 @@ class UnetModelBlock : public GGMLBlock {
         }
         if (sd_version_is_inpaint(version)) {
             in_channels = 9;
-        } else if (version == VERSION_SD1_PIX2PIX) {
+        } else if (sd_version_is_edit(version)) {
             in_channels = 8;
         }