__init__.py

from diffusers.modular_pipelines.node_utils import ModularNode



MODULE_MAP = {
    "DiffDiff": ModularNode.load_mellon_config("custom/diff-diff/diffdiff_mellon_config.json")["module"],
    "DiffDiffDenoise": ModularNode.load_mellon_config("custom/diff-diff/diffdiff_denoise_mellon_config.json")["module"],
}