diff-diff-mellon / __init__.py
YiYiXu's picture
YiYiXu HF Staff
Upload diff-diff module
ea17a8d verified
Raw
History Blame Contribute Delete
311 Bytes
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"],
}