From 2d14a03ad84d88ca27993e281fbc56aa1b9c4036 Mon Sep 17 00:00:00 2001
From: Acly <aclysia@gmail.com>
Date: Fri, 16 Jan 2026 19:28:58 +0100
Subject: [PATCH] Model inspection: detect variants of Flux 2 (Klein-4B,
 Klein-9B)

---
 api.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/api.py b/api.py
index 2e80d69..612f869 100644
--- a/api.py
+++ b/api.py
@@ -121,6 +121,9 @@ def inspect_safetensors(filename: str, model_type: str, is_checkpoint: bool):
                     raw_name = base_model.__class__.__name__
                     if raw_name == "SDXL":
                         model_type = base_model.model_type(cfg).name.lower().replace("_", "-")
+                    if raw_name == "Flux2":
+                        hidden_size = unet_config.get("hidden_size", 0)
+                        model_type = {3072: "klein-4b", 4096: "klein-9b"}.get(hidden_size, "dev")
 
             if not raw_name:
                 return {"base_model": "unknown"}