Fix Model Bug in Kontext Inference

2025-05-31 00:22:43 +03:00
parent b5c46a48e6
commit 7d4cca13cf
1 changed files with 2 additions and 1 deletions
@@ -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")