From 4d2100c6389cf434031ba36cb237ca9b57ec20b2 Mon Sep 17 00:00:00 2001
From: RITIK <ritik@Mac.lan>
Date: Thu, 27 Nov 2025 04:48:40 +0530
Subject: [PATCH] Fix pro nodes to pass base_url when creating Flux2API

---
 nodes/flux2_node.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/nodes/flux2_node.py b/nodes/flux2_node.py
index 248f0a6..16ef769 100644
--- a/nodes/flux2_node.py
+++ b/nodes/flux2_node.py
@@ -7,6 +7,7 @@ from .flux2_utils import (
     merge_reference_images,
 )
 from .flux2_utils import _blank_image_tensor as _blank_image
+from .flux2_config import Flux2Config
 
 
 def _validate_resolution(width: int, height: int) -> Optional[str]:
@@ -62,7 +63,7 @@ class Flux2ProTextToImage:
             if resolution_error:
                 return (f"Error: {resolution_error}",)
 
-            client = Flux2API()
+            client = Flux2API(base_url=Flux2Config().get_base_url())
             payload = {
                 "prompt": prompt,
                 "width": width if width > 0 else None,
@@ -183,7 +184,7 @@ class Flux2ProImageEdit:
                 "output_format": output_format,
             }
 
-            client = Flux2API()
+            client = Flux2API(base_url=Flux2Config().get_base_url())
             run_result = client.run(payload)
             image_url = run_result.get("sample")
             if not image_url: