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"