diff --git a/API_Nodes/api_caption.py b/API_Nodes/api_caption.py index 9133392..b3b6319 100644 --- a/API_Nodes/api_caption.py +++ b/API_Nodes/api_caption.py @@ -23,7 +23,7 @@ def tensor2pil(t_image: torch.Tensor) -> Image.Image: return Image.fromarray(arr) -class api_caption: +class API_caption: @classmethod def INPUT_TYPES(cls): return { @@ -193,8 +193,8 @@ class api_caption: NODE_CLASS_MAPPINGS = { - "api_caption": api_caption + "API_caption": API_caption } NODE_DISPLAY_NAME_MAPPINGS = { - "api_caption": "api_caption☀" + "API_caption": "API_caption☀" } diff --git a/API_Nodes/batch_api_caption.py b/API_Nodes/batch_api_caption.py index cc0212b..bd2fb6f 100644 --- a/API_Nodes/batch_api_caption.py +++ b/API_Nodes/batch_api_caption.py @@ -7,7 +7,7 @@ from concurrent.futures import ThreadPoolExecutor, as_completed import requests -class batch_api_caption: +class Batch_API_caption: @classmethod def INPUT_TYPES(cls): return { @@ -121,12 +121,12 @@ class batch_api_caption: if isinstance(first, dict): msg = first.get("message") or first.get("delta") or None if isinstance(msg, dict) and "content" in msg: - caption = batch_api_caption._extract_text_from_content(msg.get("content")) + caption = Batch_API_caption._extract_text_from_content(msg.get("content")) else: if "text" in first and isinstance(first.get("text"), str): caption = first.get("text") else: - caption = batch_api_caption._extract_text_from_content(first) + caption = Batch_API_caption._extract_text_from_content(first) else: caption = str(first) else: @@ -235,8 +235,8 @@ class batch_api_caption: NODE_CLASS_MAPPINGS = { - "batch_api_caption": batch_api_caption, + "Batch_API_caption": Batch_API_caption, } NODE_DISPLAY_NAME_MAPPINGS = { - "batch_api_caption": "batch_api_caption☀", + "Batch_API_caption": "Batch_API_caption☀", } diff --git a/locales/zh/nodeDefs.json b/locales/zh/nodeDefs.json index ab93513..d1a319b 100644 --- a/locales/zh/nodeDefs.json +++ b/locales/zh/nodeDefs.json @@ -1,5 +1,5 @@ { - "api_caption": { + "API_caption": { "display_name": "API 打标☀", "inputs": { "api_type": { "name": "API 类型" },