17 lines
490 B
Python
17 lines
490 B
Python
from .nodes import NODE_CLASS_MAPPINGS as CM_O, NODE_DISPLAY_NAME_MAPPINGS as NM_O
|
|
from .diffusers_warp_nodes import NODE_CLASS_MAPPINGS as CM_D, NODE_DISPLAY_NAME_MAPPINGS as NM_D
|
|
import torch
|
|
from .utils import pt_hash,pt_first_line
|
|
|
|
torch.Tensor.__hash_log__ = pt_hash
|
|
torch.Tensor.__fl_log__ = pt_first_line
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
**CM_O,
|
|
**CM_D
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
**NM_O,
|
|
**NM_D
|
|
}
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS'] |