nodes/jtrue-ComfyUI-WordEmbeddings
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source
This commit is contained in:
jtrue
2025-08-21 12:04:47 -04:00
committed by GitHub
parent 43e23e44e1
commit 49e68cf00b
1 changed files with 4 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+4 -42
View File
@@ -1,60 +1,22 @@
# __init__.py — ComfyUI-WordEmbeddings
# __init__.py — ComfyUI-WordEmbeddings (core only)
EXPERIMENTAL = True # set to False before pushing
# ---------- Core nodes ----------
from .core_nodes import (
WordEmbeddingsLoader,
WordEmbeddingsExplorer,
WordEmbeddingsInterpolator,
WordEmbeddingsEquation,
)
NODE_CLASS_MAPPINGS = {
"WordEmbeddingsLoader": WordEmbeddingsLoader,
"WordEmbeddingsExplorer": WordEmbeddingsExplorer,
"WordEmbeddingsInterpolator": WordEmbeddingsInterpolator,
"WordEmbeddingsEquation": WordEmbeddingsEquation,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"WordEmbeddingsLoader": "ComfyUI-WordEmbeddings: Loader",
"WordEmbeddingsExplorer": "ComfyUI-WordEmbeddings: Explorer",
"WordEmbeddingsInterpolator": "ComfyUI-WordEmbeddings: Interpolator",
"WordEmbeddingsEquation": "ComfyUI-WordEmbeddings: Equation",
}
# ---------- Experimental nodes (safe to skip if missing/broken) ----------
if EXPERIMENTAL:
try:
from .antonym_node import WordEmbeddingsAntonym
NODE_CLASS_MAPPINGS["WordEmbeddingsAntonym"] = WordEmbeddingsAntonym
NODE_DISPLAY_NAME_MAPPINGS["WordEmbeddingsAntonym"] = "ComfyUI-WordEmbeddings: Antonym (Experimental)"
except Exception as e:
print("[ComfyUI-WordEmbeddings] Skipping Antonym (Experimental):", e)
try:
from .analogy_node import WordEmbeddingsAnalogyEquation
NODE_CLASS_MAPPINGS["WordEmbeddingsAnalogyEquation"] = WordEmbeddingsAnalogyEquation
NODE_DISPLAY_NAME_MAPPINGS["WordEmbeddingsAnalogyEquation"] = "ComfyUI-WordEmbeddings: Analogy (Experimental)"
except Exception as e:
print("[ComfyUI-WordEmbeddings] Skipping Analogy (Experimental):", e)
try:
from .polysemy_prompt_node import WordEmbeddingsPromptPolysemy
NODE_CLASS_MAPPINGS["WordEmbeddingsPromptPolysemy"] = WordEmbeddingsPromptPolysemy
NODE_DISPLAY_NAME_MAPPINGS["WordEmbeddingsPromptPolysemy"] = "ComfyUI-WordEmbeddings: Prompt Polysemy (Experimental)"
except Exception as e:
print("[ComfyUI-WordEmbeddings] Skipping Prompt Polysemy (Experimental):", e)
try:
from .prompt_rectify_node import WordEmbeddingsPromptRectify
NODE_CLASS_MAPPINGS["WordEmbeddingsPromptRectify"] = WordEmbeddingsPromptRectify
NODE_DISPLAY_NAME_MAPPINGS["WordEmbeddingsPromptRectify"] = "ComfyUI-WordEmbeddings: Prompt Rectify (Experimental)"
except Exception as e:
print("[ComfyUI-WordEmbeddings] Skipping Prompt Rectify (Experimental):", e)
# NEW: Prompt Resolution
try:
from .prompt_resolution_node import WordEmbeddingsPromptResolution
NODE_CLASS_MAPPINGS["WordEmbeddingsPromptResolution"] = WordEmbeddingsPromptResolution
NODE_DISPLAY_NAME_MAPPINGS["WordEmbeddingsPromptResolution"] = "ComfyUI-WordEmbeddings: Prompt Resolution (Experimental)"
except Exception as e:
print("[ComfyUI-WordEmbeddings] Skipping Prompt Resolution (Experimental):", e)