diff --git a/__init__.py b/__init__.py index 77e23bf..15b3ec2 100644 --- a/__init__.py +++ b/__init__.py @@ -1,10 +1,10 @@ -from .nodes.dashscope_flux_api import DashscopeFLUXAPI +from .nodes.dashscope_flux_api import DashScopeFLUXAPI NODE_CLASS_MAPPINGS = { - "DashscopeFLUXAPI": DashscopeFLUXAPI, + "DashScopeFLUXAPI": DashScopeFLUXAPI, } NODE_DISPLAY_NAME_MAPPINGS = { - "DashscopeFLUXAPI": "FLUX API", + "DashScopeFLUXAPI": "FLUX API", } __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] diff --git a/nodes/dashscope_flux_api.py b/nodes/dashscope_flux_api.py index 99e64a6..051e512 100644 --- a/nodes/dashscope_flux_api.py +++ b/nodes/dashscope_flux_api.py @@ -6,7 +6,7 @@ from io import BytesIO -class DashscopeFLUXAPI: +class DashScopeFLUXAPI: @classmethod def INPUT_TYPES(s): return { @@ -36,7 +36,7 @@ class DashscopeFLUXAPI: RETURN_TYPES = ("IMAGE",) FUNCTION = "call_api" - CATEGORY = "Dashscope" + CATEGORY = "DashScope" def preprocess_image(self, image): image = torch.Tensor(np.array(image, dtype=np.float32) * (1 / 255)) diff --git a/workflow_examples/dashscope_flux.json b/workflow_examples/dashscope_flux.json index 51a2643..127b1b0 100644 --- a/workflow_examples/dashscope_flux.json +++ b/workflow_examples/dashscope_flux.json @@ -4,7 +4,7 @@ "nodes": [ { "id": 10, - "type": "DashscopeFLUXAPI", + "type": "DashScopeFLUXAPI", "pos": [ 426, 263 @@ -28,7 +28,7 @@ } ], "properties": { - "Node name for S&R": "DashscopeFLUXAPI" + "Node name for S&R": "DashScopeFLUXAPI" }, "widgets_values": [ "flux-schnell",