From cfa170d2b95f537d3b77e4634bfa813d6f9d40c2 Mon Sep 17 00:00:00 2001 From: "Alex \"mcmonkey\" Goodwin" Date: Wed, 17 Apr 2024 22:36:57 -0700 Subject: [PATCH] use an init file --- __init__.py | 11 +++++++++++ stability_api.py | 11 ----------- 2 files changed, 11 insertions(+), 11 deletions(-) create mode 100644 __init__.py diff --git a/__init__.py b/__init__.py new file mode 100644 index 0000000..71d70e0 --- /dev/null +++ b/__init__.py @@ -0,0 +1,11 @@ +from stability_api import StabilityCreativeUpscale, StabilityRemoveBackground, StabilityInpainting, StabilityCore, StabilitySearchAndReplace, StabilityOutpainting, StabilitySD3 + +NODE_CLASS_MAPPINGS = { + "Stability Creative Upscale": StabilityCreativeUpscale, + "Stability Remove Background": StabilityRemoveBackground, + "Stability Inpainting": StabilityInpainting, + "Stability Image Core": StabilityCore, + "Stability Search and Replace": StabilitySearchAndReplace, + "Stability Outpainting": StabilityOutpainting, + "Stability SD3": StabilitySD3, +} diff --git a/stability_api.py b/stability_api.py index b60e15b..e3c447f 100644 --- a/stability_api.py +++ b/stability_api.py @@ -258,14 +258,3 @@ class StabilityOutpainting(StabilityBase): "api_key override": ("STRING", {"multiline": False}), }, } - - -NODE_CLASS_MAPPINGS = { - "Stability Creative Upscale": StabilityCreativeUpscale, - "Stability Remove Background": StabilityRemoveBackground, - "Stability Inpainting": StabilityInpainting, - "Stability Image Core": StabilityCore, - "Stability Search and Replace": StabilitySearchAndReplace, - "Stability Outpainting": StabilityOutpainting, - "Stability SD3": StabilitySD3, -}