22 lines
679 B
Python
22 lines
679 B
Python
import os
|
|
import sys
|
|
|
|
# Add the parent directory to sys.path to allow relative imports
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
|
|
# Import node classes
|
|
from .nodes.sr_nodes import SuperResolutionModelLoader, SuperResolutionUpscale
|
|
|
|
# Define NODE_CLASS_MAPPINGS for ComfyUI
|
|
NODE_CLASS_MAPPINGS = {
|
|
"SuperResolutionModelLoader": SuperResolutionModelLoader,
|
|
"SuperResolutionUpscale": SuperResolutionUpscale,
|
|
}
|
|
|
|
# Define NODE_DISPLAY_NAME_MAPPINGS for ComfyUI
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"SuperResolutionModelLoader": "SR Model Loader",
|
|
"SuperResolutionUpscale": "SR Upscale",
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] |