nodes/Stability-AI-ComfyUI-SAI_API
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'main' into main

Browse Source
This commit is contained in:
Mohamad-Stability
2025-02-26 22:37:53 +02:00
committed by GitHub
parent 67db771e0e 504c028bfe
commit 0018d4b833
2 changed files with 22 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+4 -5
View File
@@ -1,4 +1,4 @@
from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilityImageUltra, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure, StabilitySearchAndRecolor, StabilityControlStyle, StabilityFastUpscale, StabilityReplaceRelight
from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilityImageUltra, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure, StabilityFastUpscale, StabilityControlStyle, StabilitySearchAndRecolor
NODE_CLASS_MAPPINGS = {
"Stability Conservative Upscale": StabilityConservativeUpscale,
@@ -13,8 +13,7 @@ NODE_CLASS_MAPPINGS = {
"Stability SD3": StabilitySD3,
"Stability Control Sketch": StabilityControlSketch,
"Stability Control Structure": StabilityControlStructure,
"Stability Search And Recolor": StabilitySearchAndRecolor,
"Stability Control Style": StabilityControlStyle,
"Stability Fast Upscale": StabilityFastUpscale,
"Stability Replace Background and Relight": StabilityReplaceRelight,
}
"Stability Style": StabilityControlStyle,
"Stability Search and Recolor": StabilitySearchAndRecolor,
}
stability_api.py
+18 -17
View File
@@ -267,6 +267,10 @@ class StabilityRemoveBackground(StabilityBase):
"required": {
"image": ("IMAGE",),
},
"optional": {
"output_format": (["png", "webp"],),
"api_key_override": ("STRING", {"multiline": False}),
}
}
@@ -385,7 +389,7 @@ class StabilityOutpainting(StabilityBase):
"left": ("INT", {"default": 0, "min": 0, "max": 512}),
"right": ("INT", {"default": 0, "min": 0, "max": 512}),
"up": ("INT", {"default": 0, "min": 0, "max": 512}),
"down": ("INT", {"default": 0, "min": 0, "max": 512}),\
"down": ("INT", {"default": 0, "min": 0, "max": 512}),
},
"optional": {
"prompt": ("STRING", {"multiline": True}),
@@ -431,26 +435,19 @@ class StabilityControlStructure(StabilityBase):
},
}
class StabilitySearchAndRecolor(StabilityBase):
API_ENDPOINT = "stable-image/edit/search-and-recolor"
class StabilityFastUpscale(StabilityBase):
API_ENDPOINT = "stable-image/upscale/fast"
ACCEPT = "image/*"
INPUT_SPEC = {
"required": {
"image": ("IMAGE",),
"select_prompt": ("STRING", {"multiline": True}),
"prompt": ("STRING", {"multiline": True}),
},
"optional": {
"negative_prompt": ("STRING", {"multiline": True}),
"grow_mask": ("INT", {"default": 3, "min": 0, "max": 20, "step": 1}),
"seed": ("INT", {"default": 0, "min": 0, "max": 4294967294}),
"api_key_override": ("STRING", {"multiline": False}),
"output_format": (["png", "webp", "jpeg"],),
},
"api_key_override": ("STRING", {"multiline": False}),
}
}
class StabilityControlStyle(StabilityBase):
API_ENDPOINT = "stable-image/control/style"
ACCEPT = "image/*"
@@ -469,18 +466,22 @@ class StabilityControlStyle(StabilityBase):
},
}
class StabilityFastUpscale(StabilityBase):
API_ENDPOINT = "stable-image/upscale/fast"
class StabilitySearchAndRecolor(StabilityBase):
API_ENDPOINT = "stable-image/edit/search-and-recolor"
ACCEPT = "image/*"
INPUT_SPEC = {
"required": {
"image": ("IMAGE",),
"select_prompt": ("STRING", {"multiline": True}),
"prompt": ("STRING", {"multiline": True}),
},
"optional": {
"output_format": (["png", "webp", "jpeg"],),
"negative_prompt": ("STRING", {"multiline": True}),
"grow_mask": ("INT", {"default": 3, "min": 0, "max": 20, "step": 1}),
"seed": ("INT", {"default": 0, "min": 0, "max": 4294967294}),
"api_key_override": ("STRING", {"multiline": False}),
}
"output_format": (["png", "webp", "jpeg"],),
},
}
class StabilityReplaceRelight(StabilityBase):