From 1d8849cb412bd8cc5c4516315f40b20abbe2d759 Mon Sep 17 00:00:00 2001 From: Patrick Plate Date: Fri, 10 Apr 2026 20:29:18 +0200 Subject: [PATCH] fix(mcp-image-gen): confirmed working FLUX.2 Klein encoder filename - CLIPLoader clip_name: qwen_3_4b_klein.safetensors (from Comfy-Org/vae-text-encorder-for-flux-klein-4b) - VAE: flux2-vae.safetensors (321MB, same repo) - Live test confirmed: 2.1MB photorealistic 1024x1024 PNG in 52.43s on RX 7900 XTX - Test: assert clip_name == qwen_3_4b_klein.safetensors - 37/37 tests pass --- mcp/mcp-image-gen/src/workflows/flux2_klein_heretic.json | 2 +- mcp/mcp-image-gen/tests/test_server.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/mcp/mcp-image-gen/src/workflows/flux2_klein_heretic.json b/mcp/mcp-image-gen/src/workflows/flux2_klein_heretic.json index 2bf8f32..13d0b8b 100644 --- a/mcp/mcp-image-gen/src/workflows/flux2_klein_heretic.json +++ b/mcp/mcp-image-gen/src/workflows/flux2_klein_heretic.json @@ -2,7 +2,7 @@ "1": { "class_type": "CLIPLoader", "inputs": { - "clip_name": "qwen_3_4b_bfl.safetensors", + "clip_name": "qwen_3_4b_klein.safetensors", "type": "flux2", "device": "default" } diff --git a/mcp/mcp-image-gen/tests/test_server.py b/mcp/mcp-image-gen/tests/test_server.py index 24f0184..5fe672f 100644 --- a/mcp/mcp-image-gen/tests/test_server.py +++ b/mcp/mcp-image-gen/tests/test_server.py @@ -67,6 +67,7 @@ def test_build_flux_workflow_heretic_model(): assert wf["1"]["class_type"] == "CLIPLoader" # Qwen3-4B uses single CLIPLoader assert wf["1"]["inputs"]["type"] == "flux2" # correct type for FLUX.2 assert wf["1"]["inputs"]["device"] == "default" # required for FLUX.2 CLIPLoader + assert wf["1"]["inputs"]["clip_name"] == "qwen_3_4b_klein.safetensors" # Comfy-Org/vae-text-encorder-for-flux-klein-4b assert wf["2"]["class_type"] == "CLIPTextEncode" # standard CLIP encode (not Flux-specific) assert wf["4"]["class_type"] == "UNETLoader" assert wf["4"]["inputs"]["unet_name"] == "flux-2-klein-4b.safetensors"