From 7d4cca13cfd04dd99f180d7d894de9fd4d4060d8 Mon Sep 17 00:00:00 2001
From: Jack <mohammed.aly@runware.ai>
Date: Sat, 31 May 2025 00:22:43 +0300
Subject: [PATCH] Fix Model Bug in Kontext Inference

---
 modules/runwareBFL.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/modules/runwareBFL.py b/modules/runwareBFL.py
index bf45434..3302b0f 100644
--- a/modules/runwareBFL.py
+++ b/modules/runwareBFL.py
@@ -7,7 +7,7 @@ class runwareKontext:
             "required": {
                 "Model": ([
                     "bfl:3@1 (FLUX.1 Kontext Pro)",
-                    "bfl:3@1 (FLUX.1 Kontext Max)",
+                    "bfl:4@1 (FLUX.1 Kontext Max)",
                 ], {
                     "tooltip": "Select The Model You Want For Image Generation or Image Editing.",
                     "default": "bfl:3@1 (FLUX.1 Kontext Pro)",
@@ -78,6 +78,7 @@ class runwareKontext:
 
     def kontextInference(self, **kwargs):
         runwareModel = kwargs.get("Model")
+        runwareModel = runwareModel.split(" ")[0]
         positivePrompt = kwargs.get("positivePrompt")
         promptUpsampling = kwargs.get("Prompt Upsampling")
         aspectRatio = kwargs.get("Aspect Ratio")