科林_Cyril Pilgrim
d04fff29e2
- 将所有 CATEGORY 从 DeepSeek_Toolkit 改为 🚦ComfyUI_LLMs_Toolkit
- 删除 Video File Uploader 节点 (src/utils/file_uploader.py)
- 移除 OpenAI Compatible 节点的 video_url 入参
20 lines
891 B
Python
20 lines
891 B
Python
from .src.core.llm_loader import NODE_CLASS_MAPPINGS as LLM_LOADER_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as LLM_LOADER_DISPLAY_MAPPINGS
|
|
from .src.core.openai_compatible import NODE_CLASS_MAPPINGS as OPENAI_COMPATIBLE_LOADER_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as OPENAI_COMPATIBLE_LOADER_DISPLAY_MAPPINGS
|
|
from .src.processors.image_processor import NODE_CLASS_MAPPINGS as IMAGE_PREPROCESSOR_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as IMAGE_PREPROCESSOR_DISPLAY_MAPPINGS
|
|
|
|
# 合并所有节点映射
|
|
NODE_CLASS_MAPPINGS = {
|
|
**LLM_LOADER_MAPPINGS,
|
|
**OPENAI_COMPATIBLE_LOADER_MAPPINGS,
|
|
**IMAGE_PREPROCESSOR_MAPPINGS,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**LLM_LOADER_DISPLAY_MAPPINGS,
|
|
**OPENAI_COMPATIBLE_LOADER_DISPLAY_MAPPINGS,
|
|
**IMAGE_PREPROCESSOR_DISPLAY_MAPPINGS,
|
|
}
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|