add Image Ultra API

This commit is contained in:
Alex "mcmonkey" Goodwin
2024-06-06 07:18:12 -07:00
parent 2e6b9d62b8
commit 1793086d6e
2 changed files with 19 additions and 1 deletions
+2 -1
View File
@@ -1,4 +1,4 @@
from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure
from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilityImageUltra, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure
NODE_CLASS_MAPPINGS = {
"Stability Conservative Upscale": StabilityConservativeUpscale,
@@ -7,6 +7,7 @@ NODE_CLASS_MAPPINGS = {
"Stability Erase": StabilityErase,
"Stability Inpainting": StabilityInpainting,
"Stability Image Core": StabilityCore,
"Stability Image Ultra": StabilityImageUltra,
"Stability Search and Replace": StabilitySearchAndReplace,
"Stability Outpainting": StabilityOutpainting,
"Stability SD3": StabilitySD3,
+17
View File
@@ -161,6 +161,23 @@ class StabilityCore(StabilityBase):
}
class StabilityImageUltra(StabilityBase):
API_ENDPOINT = "stable-image/generate/ultra"
ACCEPT = "image/*"
INPUT_SPEC = {
"required": {
"prompt": ("STRING", {"multiline": True}),
},
"optional": {
"negative_prompt": ("STRING", {"multiline": True}),
"seed": ("INT", {"default": 0, "min": 0, "max": 4294967294}),
"aspect_ratio": (["16:9", "1:1", "21:9", "2:3", "3:2", "4:5", "5:4", "9:16", "9:21"],),
"output_format": (["png", "jpeg"],),
"api_key_override": ("STRING", {"multiline": False}),
},
}
class StabilityConservativeUpscale(StabilityBase):
API_ENDPOINT = "stable-image/upscale/conservative"
ACCEPT = "image/*"