10 lines
306 B
Python
10 lines
306 B
Python
from .nodes.torch_compile_vae import TorchCompileVAE
|
|
from .nodes.torch_compile_controlnet import TorchCompileControlNet
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"TorchCompileVAE": TorchCompileVAE,
|
|
"TorchCompileControlNet": TorchCompileControlNet,
|
|
}
|
|
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|