diff --git a/rd_api_node.py b/rd_api_node.py index bec8a10..4ef173f 100644 --- a/rd_api_node.py +++ b/rd_api_node.py @@ -56,6 +56,7 @@ class RetroDiffusionAPINode: "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 @@ class RetroDiffusionAPINode: 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 @@ class RetroDiffusionAPINode: "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 @@ class RetroDiffusionAPINode: 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) \ No newline at end of file + return (torch.zeros(1, 64, 64, 3), 0, -1)