65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
from .clipvision_embeds import EmbedsInfo
|
|
from .clipvision_embeds import Cond2Embeds
|
|
from .clipvision_embeds import CalcEmbeds
|
|
from .clipvision_embeds import ScaleEmbeds
|
|
from .clipvision_embeds import CompareEmbeds
|
|
|
|
#from .testnodes import testnode1
|
|
|
|
from .image_search import ImageSearcher
|
|
from .image_search import ResultBrowser
|
|
from .image_search import ResultBrowserAdvanced
|
|
from .image_search import ResultCombiner
|
|
from .image_search import ResultSubtract
|
|
from .image_search import EditResults
|
|
from .image_search import FolderScores
|
|
|
|
from .clipvision_db import LoadDB
|
|
from .clipvision_db import GenerateDB
|
|
from .clipvision_db import EditDB
|
|
import folder_paths
|
|
import os
|
|
|
|
folder_paths.folder_names_and_paths["EmbDBs"] = ([os.path.join(folder_paths.models_dir, "EmbDBs")], [".json"])
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
#"testnode1": testnode1,
|
|
"EmbedsInfo": EmbedsInfo,
|
|
"Cond2Embeds": Cond2Embeds,
|
|
"ScaleEmbeds": ScaleEmbeds,
|
|
"CalcEmbeds": CalcEmbeds,
|
|
"CompareEmbeds": CompareEmbeds,
|
|
"ImageSearcher": ImageSearcher,
|
|
"ResultBrowser": ResultBrowser,
|
|
"ResultBrowserAdvanced": ResultBrowserAdvanced,
|
|
"ResultCombiner": ResultCombiner,
|
|
"ResultSubtract": ResultSubtract,
|
|
"EditResults": EditResults,
|
|
"LoadDB": LoadDB,
|
|
"GenerateDB": GenerateDB,
|
|
"EditDB": EditDB,
|
|
"FolderScores": FolderScores,
|
|
}
|
|
|
|
# A dictionary that contains the friendly/humanly readable titles for the nodes
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
#"testnode1": "Testnode (experimental)",
|
|
"EmbedsInfo": "Embeddings Shape Info (experimental)",
|
|
"ScaleEmbeds": "Scale Embeddings (experimental)",
|
|
"CalcEmbeds": "Calculate Embeddings (experimental)",
|
|
"Cond2Embeds": "Condition 2 Embeddings",
|
|
"CompareEmbeds": "Compare Embeddings",
|
|
"ImageSearcher": "Image Searcher",
|
|
"ResultBrowser": "Result Browser",
|
|
"ResultBrowserAdvanced": "Advanced Result Browser",
|
|
"ResultCombiner": "Combine Results",
|
|
"ResultSubtract": "Subtract Results",
|
|
"EditResults": "Edit Results",
|
|
"DBLoader": "Load Embeddings Database",
|
|
"GenerateDB": "Generate Embeddings Database",
|
|
"EditDB": "Edit Embeddings Database",
|
|
"FolderScores": "Folder Scores",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|