Files
2025-02-27 09:39:24 +08:00

21 lines
1.3 KiB
Python

from .stability_api import StabilityConservativeUpscale, StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityErase, StabilityCore, StabilityImageUltra, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3, StabilityControlSketch, StabilityControlStructure, StabilitySearchAndRecolor, StabilityControlStyle, StabilityFastUpscale, StabilityReplaceRelight
NODE_CLASS_MAPPINGS = {
"Stability Conservative Upscale": StabilityConservativeUpscale,
"Stability Creative Upscale": StabilityCreativeUpscale,
"Stability Remove Background": StabilityRemoveBackground,
"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,
"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,
}