| from .pag_nodes import PerturbedAttention, SmoothedEnergyGuidanceAdvanced | |
| from .pag_trt_nodes import TRTAttachPag, TRTPerturbedAttention | |
| NODE_CLASS_MAPPINGS = { | |
| "PerturbedAttention": PerturbedAttention, | |
| "SmoothedEnergyGuidanceAdvanced": SmoothedEnergyGuidanceAdvanced, | |
| "TRTAttachPag": TRTAttachPag, | |
| "TRTPerturbedAttention": TRTPerturbedAttention, | |
| } | |
| NODE_DISPLAY_NAME_MAPPINGS = { | |
| "PerturbedAttention": "Perturbed-Attention Guidance (Advanced)", | |
| "SmoothedEnergyGuidanceAdvanced": "Smoothed Energy Guidance (Advanced)", | |
| "TRTAttachPag": "TensorRT Attach PAG", | |
| "TRTPerturbedAttention": "TensorRT Perturbed-Attention Guidance", | |
| } | |