18 lines
731 B
Python
18 lines
731 B
Python
from .py import nodes
|
|
from .py import custom_noise
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"OCS Sampler": nodes.SamplerNode,
|
|
"OCS Substeps": nodes.SubstepsNode,
|
|
"OCS Group": nodes.GroupNode,
|
|
"OCS Param": nodes.ParamNode,
|
|
"OCS MultiParam": nodes.MultiParamNode,
|
|
"OCS ModelSetMaxSigma": nodes.ModelSetMaxSigmaNode,
|
|
"OCS SimpleRestartSchedule": nodes.SimpleRestartSchedule,
|
|
"OCS ApplyFilterLatent": nodes.ApplyFilterLatent,
|
|
"OCS ApplyFilterImage": nodes.ApplyFilterImage,
|
|
"OCS ExpressionFilteredLatentOperation": nodes.ExpressionFilteredLatentOperationNode,
|
|
"OCS ExpressionFilteredModelPatch": nodes.ExpressionFilteredModelPatchNode,
|
|
} | custom_noise.NODE_CLASS_MAPPINGS
|
|
__all__ = ["NODE_CLASS_MAPPINGS"]
|