Vikramjeet
941c722c7f
- Added structured logging - Removed SingleGPUSaverNode and MultiGPUSaverNode classes - Added VideoSaverNode class to handle video saving - Updated imports and mappings in __init__.py #refactor #codestructure #classnames #nodes
29 lines
1.1 KiB
Python
29 lines
1.1 KiB
Python
"""VEnhancer nodes for ComfyUI."""
|
|
|
|
from .nodes.single_gpu_loader import SingleGPUVEnhancerLoader
|
|
from .nodes.multi_gpu_loader import MultiGPUVEnhancerLoader
|
|
from .nodes.single_gpu_inference import SingleGPUInferenceNode
|
|
from .nodes.multigpu_inference import MultiGPUInferenceNode
|
|
from .nodes.video_saver import VideoSaverNode
|
|
from .nodes.video_loader import VideoLoaderNode
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SingleGPUVEnhancerLoader": SingleGPUVEnhancerLoader,
|
|
"MultiGPUVEnhancerLoader": MultiGPUVEnhancerLoader,
|
|
"SingleGPUInference": SingleGPUInferenceNode,
|
|
"MultiGPUInference": MultiGPUInferenceNode,
|
|
"VideoSaver": VideoSaverNode,
|
|
"VideoLoader": VideoLoaderNode,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SingleGPUVEnhancerLoader": "Load VEnhancer (Single GPU)",
|
|
"MultiGPUVEnhancerLoader": "Load VEnhancer (Multi-GPU)",
|
|
"SingleGPUInference": "Enhance Video (Single GPU)",
|
|
"MultiGPUInference": "Enhance Video (Multi-GPU)",
|
|
"VideoSaver": "Save Video",
|
|
"VideoLoader": "Load Video",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] |