nodes/svntax-ComfyUI-RetroDiffusion-API-Node
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added bypass_prompt_expansion

Browse Source
This commit is contained in:
svntax
2025-09-29 07:33:26 -04:00
parent 8677264c52
commit 729077eef1
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
rd_api_node.py
+4 -2
View File
@@ -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)
return (torch.zeros(1, 64, 64, 3), 0, -1)