From 4532038c8a0cfc365ce3c81f1d2aabdc851ab859 Mon Sep 17 00:00:00 2001
From: Vito Sansevero <vito.sansevero@gmail.com>
Date: Wed, 28 Oct 2026 07:12:29 -0700
Subject: [PATCH] feat(model_resolution_picker): add a Krea 2 family

Krea 2 uses the Wan 2.1 latent (8x VAE) with patch 2, so sizes sit on a
16 px grid. Tiers 1.0 / 2.0 / 4.0 MP cover the model card's 1024x1024 and
2048x2048; Big Love KreaEdit3's 1024x1536 is listed under 2:3.
---
 CHANGELOG.md                                  |   8 +
 .../documentation/model_resolution_picker.md  |   1 +
 .../model_resolution_picker/families.json     | 182 ++++++++++++++++++
 .../model_resolution_picker/test_families.py  |   1 +
 4 files changed, 192 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 19265b4..89e711d 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,5 +1,13 @@
 # Changelog
 
+## Unreleased
+
+### Model Resolution Picker
+
+- New `Krea 2` family (Krea 2 Turbo, Raw and finetunes): 8x VAE, step 16,
+  1.0 / 2.0 / 4.0 MP tiers, with the model card's 1024×1024 and 2048×2048 and
+  Big Love KreaEdit3's 1024×1536.
+
 ## 1.3.0
 
 ### Model Resolution Picker
diff --git a/examples/documentation/model_resolution_picker.md b/examples/documentation/model_resolution_picker.md
index 4e3aaf1..7a36be8 100644
--- a/examples/documentation/model_resolution_picker.md
+++ b/examples/documentation/model_resolution_picker.md
@@ -44,6 +44,7 @@ The frontend reads the family table from `GET /kikotools/model_resolution/famili
 | Z-Image | Z-Image Turbo and Z-Image Base | 8x | 16 | 1.0, 1.6 | 1024×1024 | image |
 | Qwen-Image | Qwen-Image and Qwen-Image 2512 | 8x | 16 | 1.76 | 1328×1328 | image |
 | Flux.2 / Klein / MOSS | Flux.2 dev, Flux.2 Klein, Flux.3 Klein, MOSS-Image, Ideogram 4 | 16x | 16 | 1.0, 2.0, 4.0 | 1024×1024 | image |
+| Krea 2 | Krea 2 Turbo, Krea 2 Raw and Krea 2 finetunes (Big Love KreaEdit3) | 8x | 16 | 1.0, 2.0, 4.0 | 1024×1024 | image |
 | Qwen-Image 2.1 | Qwen-Image 2.1 and 2.1 Turbo | 16x | 32 | 1.0, 4.0 | 1024×1024 | image |
 | HunyuanImage 2.1 | HunyuanImage 2.1 | 32x | 32 | 4.0 | 2048×2048 | image |
 | Wan 2.2 (keyframes) | Wan 2.2 (first/last-frame sizes) | 16x | 16 | 0.4, 0.9 | 1280×720 | video |
diff --git a/kikotools/tools/model_resolution_picker/families.json b/kikotools/tools/model_resolution_picker/families.json
index 6fb9760..426f2ca 100644
--- a/kikotools/tools/model_resolution_picker/families.json
+++ b/kikotools/tools/model_resolution_picker/families.json
@@ -1107,6 +1107,188 @@
       },
       "notes": "generated at tiers 1.0, 2.0, 4.0 MP, step 16; 16:9 2.0 MP set to 1920x1088"
     },
+    {
+      "label": "Krea 2",
+      "aliases": [],
+      "models": "Krea 2 Turbo, Krea 2 Raw and Krea 2 finetunes (Big Love KreaEdit3)",
+      "group": "image",
+      "vaeRatio": 8,
+      "step": 16,
+      "megapixels": [
+        1.0,
+        2.0,
+        4.0
+      ],
+      "default": [
+        1024,
+        1024
+      ],
+      "sizes": {
+        "1:1": [
+          [
+            1024,
+            1024
+          ],
+          [
+            1456,
+            1456
+          ],
+          [
+            2048,
+            2048
+          ]
+        ],
+        "4:5": [
+          [
+            912,
+            1152
+          ],
+          [
+            1296,
+            1616
+          ],
+          [
+            1824,
+            2288
+          ]
+        ],
+        "5:4": [
+          [
+            1152,
+            912
+          ],
+          [
+            1616,
+            1296
+          ],
+          [
+            2288,
+            1824
+          ]
+        ],
+        "3:4": [
+          [
+            880,
+            1184
+          ],
+          [
+            1248,
+            1680
+          ],
+          [
+            1776,
+            2368
+          ]
+        ],
+        "4:3": [
+          [
+            1184,
+            880
+          ],
+          [
+            1680,
+            1248
+          ],
+          [
+            2368,
+            1776
+          ]
+        ],
+        "2:3": [
+          [
+            832,
+            1248
+          ],
+          [
+            1024,
+            1536
+          ],
+          [
+            1184,
+            1776
+          ],
+          [
+            1680,
+            2512
+          ]
+        ],
+        "3:2": [
+          [
+            1248,
+            832
+          ],
+          [
+            1536,
+            1024
+          ],
+          [
+            1776,
+            1184
+          ],
+          [
+            2512,
+            1680
+          ]
+        ],
+        "9:16": [
+          [
+            768,
+            1360
+          ],
+          [
+            1088,
+            1936
+          ],
+          [
+            1536,
+            2736
+          ]
+        ],
+        "16:9": [
+          [
+            1360,
+            768
+          ],
+          [
+            1936,
+            1088
+          ],
+          [
+            2736,
+            1536
+          ]
+        ],
+        "21:9": [
+          [
+            1568,
+            672
+          ],
+          [
+            2208,
+            944
+          ],
+          [
+            3136,
+            1344
+          ]
+        ],
+        "9:21": [
+          [
+            672,
+            1568
+          ],
+          [
+            944,
+            2208
+          ],
+          [
+            1344,
+            3136
+          ]
+        ]
+      },
+      "notes": "Model card examples are 1024x1024 and 2048x2048; generated at tiers 1.0, 2.0, 4.0 MP, step 16 (Wan 2.1 latent, patch 2). 1024x1536 is Big Love KreaEdit3's model page size."
+    },
     {
       "label": "Qwen-Image 2.1",
       "aliases": [],
diff --git a/tests/unit/tools/model_resolution_picker/test_families.py b/tests/unit/tools/model_resolution_picker/test_families.py
index 38e14b8..6aa6f6f 100644
--- a/tests/unit/tools/model_resolution_picker/test_families.py
+++ b/tests/unit/tools/model_resolution_picker/test_families.py
@@ -71,6 +71,7 @@ def test_defaults_labels_and_must_includes():
         "SDXL": [(1344, 768)],
         "Flux.1": [(1920, 1088), (1088, 1920)],
         "Flux.2 / Klein / MOSS": flux2,
+        "Krea 2": [(1024, 1024), (2048, 2048), (1024, 1536), (1536, 1024)],
         "Z-Image": [(1680, 720), (1280, 1280)],
         "Qwen-Image": [(1664, 928)],
         "Qwen-Image 2.1": [(2048, 2048), (2400, 1792), (2528, 1696), (2752, 1536)],