499096bdb0
* Added Super Resolution to the streampack * Super Resolution created by ryan, updated for missing * add prestartup script to check for opencv-cuda --------- Co-authored-by: ryanontheinside <7623207+ryanontheinside@users.noreply.github.com> Co-authored-by: John | Elite Encoder <john@eliteencoder.net>
16 lines
550 B
Python
16 lines
550 B
Python
# Import node classes
|
|
from ..src.super_resolution.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"] |