add SDXL refiner

scripts/toolkit_gui.py

@@ -300,7 +300,8 @@ def do_adv_report(details: ToolkitModel, abbreviate=True):
                     data = data[:5]
                 for k in data:
                     if type(k) == tuple:
-                        report += f"  - {k[0]} {k[1]}\n"
+                        size = str(list(k[1])).replace(" ", "")
+                        report += f"  - {k[0]} {size}\n"
                     else:
                         report += f"  - {k}\n"
                 if will_abbreviate:
@@ -315,7 +316,8 @@ def do_adv_report(details: ToolkitModel, abbreviate=True):
         if will_abbreviate:
             data = data[5:]
         for k, z in data:
-            report += f" - {k} {z}\n"
+            size = str(list(z)).replace(" ", "")
+            report += f" - {k} {size}\n"
         if will_abbreviate:
             report += f" - ...\n"
 

toolkit.py

@@ -71,6 +71,11 @@ COMPONENTS = {
         "source": "UNET-XL-SD.txt",
         "prefix": "model.diffusion_model."
     },
+    "UNET-XL-Refiner": {
+        "keys": {},
+        "source": "UNET-XL-Refiner.txt",
+        "prefix": "model.diffusion_model."
+    },
     "VAE-v1-SD": {
         "keys": {},
         "source": "VAE-v1-SD.txt",
@@ -101,6 +106,11 @@ COMPONENTS = {
         "source": "CLIP-XL-SD.txt",
         "prefix": "conditioner.embedders.1.model."
     },
+    "CLIP-XL-Refiner": {
+        "keys": {},
+        "source": "CLIP-XL-SD.txt",
+        "prefix": "conditioner.embedders.0.model."
+    },
     "CLIP-XL-AUX-SD": {
         "keys": {},
         "source": "CLIP-v1-SD.txt",
@@ -153,12 +163,14 @@ COMPONENT_CLASS = {
     "UNET-v2-Inpainting": "UNET-v2",
     "UNET-v2-Depth": "UNET-v2-Depth",
     "UNET-XL-SD": "UNET-XL",
+    "UNET-XL-Refiner": "UNET-XL-Refiner",
     "VAE-v1-SD": "VAE-v1",
     "CLIP-v1-SD": "CLIP-v1",
     "CLIP-v1-NAI": "CLIP-v1",
     "CLIP-v2-SD": "CLIP-v2",
     "CLIP-v2-WD": "CLIP-v2",
     "CLIP-XL-SD": "CLIP-XL",
+    "CLIP-XL-Refiner": "CLIP-XL",
     "CLIP-XL-AUX-SD": "CLIP-XL-AUX",
     "Depth-v2-SD": "Depth-v2",
     "LoRA-v1-UNET": "LoRA-v1-UNET",
@@ -217,6 +229,12 @@ ARCHITECTURES = {
         "required": [],
         "prefixed": False
     },
+    "UNET-XL-Refiner": {
+        "classes": ["UNET-XL-Refiner"],
+        "optional": [],
+        "required": [],
+        "prefixed": False
+    },
     "VAE-v1": {
         "classes": ["VAE-v1"],
         "optional": [],
@@ -291,7 +309,13 @@ ARCHITECTURES = {
     },
     "SD-XL": {
         "classes": ["UNET-XL", "VAE-v1", "CLIP-XL", "CLIP-XL-AUX"],
-        "optional": OPTIONAL + ["conditioner.embedders.0.transformer.text_model.embeddings.position_ids"],
+        "optional": OPTIONAL,
+        "required": [],
+        "prefixed": True
+    },
+    "SD-XL-Refiner": {
+        "classes": ["UNET-XL-Refiner", "VAE-v1", "CLIP-XL"],
+        "optional": OPTIONAL,
         "required": [],
         "prefixed": True
     },