Added bypass_prompt_expansion

Author: svntax

rd_api_node.py

@@ -56,6 +56,7 @@ def INPUT_TYPES(cls):
                 "width": ("INT", {"default": 128, "min": 16, "max": 512, "step": 8}),
                 "height": ("INT", {"default": 128, "min": 16, "max": 512, "step": 8}),
                 "prompt_style": (prompt_styles,),
+                "bypass_prompt_expansion": ("BOOLEAN", {"default": False}),
                 "remove_bg": ("BOOLEAN", {"default": False}),
                 "tile_x": ("BOOLEAN", {"default": False}),
                 "tile_y": ("BOOLEAN", {"default": False}),
@@ -74,7 +75,7 @@ def INPUT_TYPES(cls):
     FUNCTION = "generate_image"
     CATEGORY = "RetroDiffusion"
 
-    def generate_image(self, seed, width, height, prompt_style, remove_bg, tile_x, tile_y, return_spritesheet_for_animations, prompt, denoise, input_image=None):
+    def generate_image(self, seed, width, height, prompt_style, bypass_prompt_expansion, remove_bg, tile_x, tile_y, return_spritesheet_for_animations, prompt, denoise, input_image=None):
         url = "https://api.retrodiffusion.ai/v1/inferences"
 
         headers = {
@@ -86,6 +87,7 @@ def generate_image(self, seed, width, height, prompt_style, remove_bg, tile_x, t
             "width": width,
             "height": height,
             "prompt_style": prompt_style,
+            "bypass_prompt_expansion": bypass_prompt_expansion,
             "remove_bg": remove_bg,
             "tile_x": tile_x,
             "tile_y": tile_y,
@@ -127,4 +129,4 @@ def generate_image(self, seed, width, height, prompt_style, remove_bg, tile_x, t
         except Exception as e:
             print(f"Error processing response: {e}")
             # Return empty tensor and 0 cost on error
-            return (torch.zeros(1, 64, 64, 3), 0, -1)
+            return (torch.zeros(1, 64, 64, 3), 0, -1)