41 lines
1.7 KiB
Python
41 lines
1.7 KiB
Python
# __init__.py — ComfyUI-WordEmbeddings
|
|
|
|
from .core_nodes import (
|
|
WordEmbeddingsLoader,
|
|
WordEmbeddingsLocalModelLoader,
|
|
WordEmbeddingsExplorer,
|
|
WordEmbeddingsInterpolator,
|
|
WordEmbeddingsEquation,
|
|
WordEmbeddingsTokenNeighbors,
|
|
WordEmbeddingsTokenAxis,
|
|
WordEmbeddingsTokenAxis2D,
|
|
WordEmbeddingsTokenAxis3D,
|
|
WordEmbeddingsTokenCentrality,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"WordEmbeddingsLoader": WordEmbeddingsLoader,
|
|
"WordEmbeddingsLocalModelLoader": WordEmbeddingsLocalModelLoader,
|
|
"WordEmbeddingsExplorer": WordEmbeddingsExplorer,
|
|
"WordEmbeddingsInterpolator": WordEmbeddingsInterpolator,
|
|
"WordEmbeddingsEquation": WordEmbeddingsEquation,
|
|
"WordEmbeddingsTokenNeighbors": WordEmbeddingsTokenNeighbors,
|
|
"WordEmbeddingsTokenAxis": WordEmbeddingsTokenAxis,
|
|
"WordEmbeddingsTokenAxis2D": WordEmbeddingsTokenAxis2D,
|
|
"WordEmbeddingsTokenAxis3D": WordEmbeddingsTokenAxis3D,
|
|
"WordEmbeddingsTokenCentrality": WordEmbeddingsTokenCentrality,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"WordEmbeddingsLoader": "WordEmbeddings: Loader",
|
|
"WordEmbeddingsLocalModelLoader": "WordEmbeddings: Local Loader",
|
|
"WordEmbeddingsExplorer": "WordEmbeddings: Explorer",
|
|
"WordEmbeddingsInterpolator": "WordEmbeddings: Interpolator",
|
|
"WordEmbeddingsEquation": "WordEmbeddings: Equation",
|
|
"WordEmbeddingsTokenNeighbors": "WordEmbeddings: Token Neighbors",
|
|
"WordEmbeddingsTokenAxis": "WordEmbeddings: Token Axis",
|
|
"WordEmbeddingsTokenAxis2D": "WordEmbeddings: Token Axis2D",
|
|
"WordEmbeddingsTokenAxis3D": "WordEmbeddings: Token Axis3D",
|
|
"WordEmbeddingsTokenCentrality": "WordEmbeddings: Token Centrality",
|
|
}
|