__init__.py

import traceback 
import sys 

NODE_CLASS_MAPPINGS = {}

# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {}

try:
    from .node import ApplyStableFastUnet

    SF_NODE_CLASS_MAPPINGS = {
        "ApplyStableFastUnet": ApplyStableFastUnet,
    }

    SF_NODE_DISPLAY_NAME_MAPPINGS = {
        "ApplyStableFastUnet": "Apply StableFast Unet",
    }
    NODE_CLASS_MAPPINGS.update(SF_NODE_CLASS_MAPPINGS)
    NODE_DISPLAY_NAME_MAPPINGS.update(SF_NODE_DISPLAY_NAME_MAPPINGS)
except Exception as e:
    print("ComfyUI_stable_fast: StableFast node import failed.")
    traceback.print_exception(*sys.exc_info()) 

try:
    from .tensorrt_node import (
        ApplyTensorRTControlNet,
        ApplyTensorRTUnet,
        ApplyTensorRTVaeDecoder,
    )

    TRT_NODE_CLASS_MAPPINGS = {
        "ApplyTensorRTUnet": ApplyTensorRTUnet,
        "ApplyTensorRTVaeDecoder": ApplyTensorRTVaeDecoder,
        "ApplyTensorRTControlNet": ApplyTensorRTControlNet,
    }
    TRT_NODE_DISPLAY_NAME_MAPPINGS = {
        "ApplyTensorRTUnet": "Apply TensorRT Unet",
        "ApplyTensorRTVaeDecoder": "Apply TensorRT VaeDecoder",
        "ApplyTensorRTControlNet": "Apply TensorRT ControlNet",
    }
    NODE_CLASS_MAPPINGS.update(TRT_NODE_CLASS_MAPPINGS)
    NODE_DISPLAY_NAME_MAPPINGS.update(TRT_NODE_DISPLAY_NAME_MAPPINGS)
except Exception as e:
    print("ComfyUI_stable_fast: tensorrt_node import failed.")
    traceback.print_exception(*sys.exc_info()) 

if len(NODE_CLASS_MAPPINGS) == 0:
    raise Exception("import failed")