22 lines
855 B
Python
22 lines
855 B
Python
import sys, os
|
|
sys.path.append(os.path.dirname(__file__))
|
|
from .node import (ResetModelPatcherCalculateWeight,
|
|
MakeModelMemorySafe,
|
|
SimpleDummyModel,
|
|
SimpleDummyRun)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ResetModelPatcher-safewrapper": ResetModelPatcherCalculateWeight,
|
|
"MakeModelMemorySafe-safewrapper": MakeModelMemorySafe,
|
|
"SimpleDummyModel-safewrapper": SimpleDummyModel,
|
|
"SimpleDummyRun-safewrapper": SimpleDummyRun,
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ResetModelPatcher-safewrapper": "Reset model patcher (middlek)",
|
|
"MakeModelMemorySafe-safewrapper": "Memory safe wrap (middlek)",
|
|
"SimpleDummyModel-safewrapper": "Simple dummy model (middlek)",
|
|
"SimpleDummyRun-safewrapper": "Simpler dummy run (middlek)",
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|