356 lines
14 KiB
Python
356 lines
14 KiB
Python
# This file is auto-generated. Do not edit manually.
|
|
from typing import Dict, Type, TypeVar
|
|
from .modules.pytorch_wrapper.pt_abs import PtAbs
|
|
from .modules.pytorch_wrapper.pt_acos import PtAcos
|
|
from .modules.pytorch_wrapper.pt_add import PtAdd
|
|
from .modules.pytorch_wrapper.pt_arange import PtArange
|
|
from .modules.pytorch_wrapper.pt_argmax import PtArgmax
|
|
from .modules.pytorch_wrapper.pt_argmin import PtArgmin
|
|
from .modules.pytorch_wrapper.pt_asin import PtAsin
|
|
from .modules.pytorch_wrapper.pt_atan import PtAtan
|
|
from .modules.pytorch_wrapper.pt_bitwise_and import PtBitwiseAnd
|
|
from .modules.pytorch_wrapper.pt_bitwise_left_shift import PtBitwiseLeftShift
|
|
from .modules.pytorch_wrapper.pt_bitwise_not import PtBitwiseNot
|
|
from .modules.pytorch_wrapper.pt_bitwise_or import PtBitwiseOr
|
|
from .modules.pytorch_wrapper.pt_bitwise_right_shift import PtBitwiseRightShift
|
|
from .modules.pytorch_wrapper.pt_bitwise_xor import PtBitwiseXor
|
|
from .modules.pytorch_wrapper.pt_bmm import PtBmm
|
|
from .modules.pytorch_wrapper.pt_bool_create import PtBoolCreate
|
|
from .modules.pytorch_wrapper.pt_cos import PtCos
|
|
from .modules.pytorch_wrapper.pt_cosh import PtCosh
|
|
from .modules.pytorch_wrapper.pt_crop import PtCrop
|
|
from .modules.pytorch_wrapper.pt_data_loader import PtDataLoader
|
|
from .modules.pytorch_wrapper.pt_div import PtDiv
|
|
from .modules.pytorch_wrapper.pt_einsum import PtEinsum
|
|
from .modules.pytorch_wrapper.pt_eq import PtEq
|
|
from .modules.pytorch_wrapper.pt_evaluate_classification_model import PtEvaluateClassificationModel
|
|
from .modules.pytorch_wrapper.pt_exp import PtExp
|
|
from .modules.pytorch_wrapper.pt_flatten import PtFlatten
|
|
from .modules.pytorch_wrapper.pt_float_create import PtFloatCreate
|
|
from .modules.pytorch_wrapper.pt_floor_divide import PtFloorDiv
|
|
from .modules.pytorch_wrapper.pt_from_image import PtFromImage
|
|
from .modules.pytorch_wrapper.pt_from_image_transpose import PtFromImageTranspose
|
|
from .modules.pytorch_wrapper.pt_from_latent import PtFromLatent
|
|
from .modules.pytorch_wrapper.pt_from_numpy import PtFromNumpy
|
|
from .modules.pytorch_wrapper.pt_full import PtFull
|
|
from .modules.pytorch_wrapper.pt_gather import PtGather
|
|
from .modules.pytorch_wrapper.pt_ge import PtGe
|
|
from .modules.pytorch_wrapper.pt_gt import PtGt
|
|
from .modules.pytorch_wrapper.pt_index_select import PtIndexSelect
|
|
from .modules.pytorch_wrapper.pt_int_create import PtIntCreate
|
|
from .modules.pytorch_wrapper.pt_interpolate_by_scale_factor import PtInterpolateByScaleFactor
|
|
from .modules.pytorch_wrapper.pt_interpolate_to_size import PtInterpolateToSize
|
|
from .modules.pytorch_wrapper.pt_le import PtLe
|
|
from .modules.pytorch_wrapper.pt_linspace import PtLinspace
|
|
from .modules.pytorch_wrapper.pt_load_model import PtLoadModel
|
|
from .modules.pytorch_wrapper.pt_log import PtLog
|
|
from .modules.pytorch_wrapper.pt_logical_and import PtLogicalAnd
|
|
from .modules.pytorch_wrapper.pt_logical_not import PtLogicalNot
|
|
from .modules.pytorch_wrapper.pt_logical_or import PtLogicalOr
|
|
from .modules.pytorch_wrapper.pt_logical_xor import PtLogicalXor
|
|
from .modules.pytorch_wrapper.pt_lt import PtLt
|
|
from .modules.pytorch_wrapper.pt_masked_select import PtMaskedSelect
|
|
from .modules.pytorch_wrapper.pt_matmul import PtMatMul
|
|
from .modules.pytorch_wrapper.pt_max import PtMax
|
|
from .modules.pytorch_wrapper.pt_mean import PtMean
|
|
from .modules.pytorch_wrapper.pt_median import PtMedian
|
|
from .modules.pytorch_wrapper.pt_min import PtMin
|
|
from .modules.pytorch_wrapper.pt_mm import PtMm
|
|
from .modules.pytorch_wrapper.pt_mul import PtMul
|
|
from .modules.pytorch_wrapper.pt_ne import PtNe
|
|
from .modules.pytorch_wrapper.pt_neg import PtNeg
|
|
from .modules.pytorch_wrapper.pt_ones import PtOnes
|
|
from .modules.pytorch_wrapper.pt_pad import PtPad
|
|
from .modules.pytorch_wrapper.pt_permute import PtPermute
|
|
from .modules.pytorch_wrapper.pt_pow import PtPow
|
|
from .modules.pytorch_wrapper.pt_predict_classification_model import PtPredictClassificationModel
|
|
from .modules.pytorch_wrapper.pt_prod import PtProd
|
|
from .modules.pytorch_wrapper.pt_rand import PtRand
|
|
from .modules.pytorch_wrapper.pt_rand_int import PtRandInt
|
|
from .modules.pytorch_wrapper.pt_randn import PtRandn
|
|
from .modules.pytorch_wrapper.pt_remainder import PtRemainder
|
|
from .modules.pytorch_wrapper.pt_reshape import PtReshape
|
|
from .modules.pytorch_wrapper.pt_save_model import PtSaveModel
|
|
from .modules.pytorch_wrapper.pt_scatter import PtScatter
|
|
from .modules.pytorch_wrapper.pt_show_size import PtShowSize
|
|
from .modules.pytorch_wrapper.pt_show_text import PtShowText
|
|
from .modules.pytorch_wrapper.pt_sin import PtSin
|
|
from .modules.pytorch_wrapper.pt_sinh import PtSinh
|
|
from .modules.pytorch_wrapper.pt_size import PtSize
|
|
from .modules.pytorch_wrapper.pt_size_create import PtSizeCreate
|
|
from .modules.pytorch_wrapper.pt_size_to_numpy import PtSizeToNumpy
|
|
from .modules.pytorch_wrapper.pt_size_to_string import PtSizeToString
|
|
from .modules.pytorch_wrapper.pt_sqrt import PtSqrt
|
|
from .modules.pytorch_wrapper.pt_squeeze import PtSqueeze
|
|
from .modules.pytorch_wrapper.pt_std import PtStd
|
|
from .modules.pytorch_wrapper.pt_sub import PtSub
|
|
from .modules.pytorch_wrapper.pt_sum import PtSum
|
|
from .modules.pytorch_wrapper.pt_tan import PtTan
|
|
from .modules.pytorch_wrapper.pt_tanh import PtTanh
|
|
from .modules.pytorch_wrapper.pt_to_image import PtToImage
|
|
from .modules.pytorch_wrapper.pt_to_image_transpose import PtToImageTranspose
|
|
from .modules.pytorch_wrapper.pt_to_latent import PtToLatent
|
|
from .modules.pytorch_wrapper.pt_to_numpy import PtToNumpy
|
|
from .modules.pytorch_wrapper.pt_to_rgb_tensors import PtToRgbTensors
|
|
from .modules.pytorch_wrapper.pt_train_classification_model import PtTrainClassificationModel
|
|
from .modules.pytorch_wrapper.pt_unsqueeze import PtUnsqueeze
|
|
from .modules.pytorch_wrapper.pt_var import PtVar
|
|
from .modules.pytorch_wrapper.pt_view import PtView
|
|
from .modules.pytorch_wrapper.pt_where import PtWhere
|
|
from .modules.pytorch_wrapper.pt_zeros import PtZeros
|
|
from .modules.pytorch_wrapper.ptn_conv_model import PtnConvModel
|
|
from .modules.pytorch_wrapper.ptn_linear_model import PtnLinearModel
|
|
from .modules.pytorch_wrapper.ptn_resnet_model import PtnResnetModel
|
|
from .modules.pytorch_wrapper.pto_adam import PtoAdam
|
|
from .modules.pytorch_wrapper.ptv_dataset import PtvDataset
|
|
from .modules.pytorch_wrapper.ptv_dataset_len import PtvDatasetLen
|
|
from .modules.pytorch_wrapper.ptv_dataset_loader import PtvDatasetLoader
|
|
from .modules.pytorch_wrapper.ptv_image_folder_dataset import PtvImageFolderDataset
|
|
from .modules.pytorch_wrapper.ptv_transforms_resize import PtvTransformsResize
|
|
from .modules.pytorch_wrapper.ptv_transforms_to_tensor import PtvTransformsToTensor
|
|
T = TypeVar("T")
|
|
|
|
|
|
"""
|
|
NODE_CLASS_MAPPINGS (Dict[str, Type[T]]):
|
|
A dictionary mapping node names to their corresponding class implementations.
|
|
"""
|
|
|
|
NODE_CLASS_MAPPINGS: Dict[str, Type[T]] = {
|
|
"PtAbs": PtAbs,
|
|
"PtAcos": PtAcos,
|
|
"PtAdd": PtAdd,
|
|
"PtArange": PtArange,
|
|
"PtArgmax": PtArgmax,
|
|
"PtArgmin": PtArgmin,
|
|
"PtAsin": PtAsin,
|
|
"PtAtan": PtAtan,
|
|
"PtBitwiseAnd": PtBitwiseAnd,
|
|
"PtBitwiseLeftShift": PtBitwiseLeftShift,
|
|
"PtBitwiseNot": PtBitwiseNot,
|
|
"PtBitwiseOr": PtBitwiseOr,
|
|
"PtBitwiseRightShift": PtBitwiseRightShift,
|
|
"PtBitwiseXor": PtBitwiseXor,
|
|
"PtBmm": PtBmm,
|
|
"PtBoolCreate": PtBoolCreate,
|
|
"PtCos": PtCos,
|
|
"PtCosh": PtCosh,
|
|
"PtCrop": PtCrop,
|
|
"PtDataLoader": PtDataLoader,
|
|
"PtDiv": PtDiv,
|
|
"PtEinsum": PtEinsum,
|
|
"PtEq": PtEq,
|
|
"PtEvaluateClassificationModel": PtEvaluateClassificationModel,
|
|
"PtExp": PtExp,
|
|
"PtFlatten": PtFlatten,
|
|
"PtFloatCreate": PtFloatCreate,
|
|
"PtFloorDiv": PtFloorDiv,
|
|
"PtFromImage": PtFromImage,
|
|
"PtFromImageTranspose": PtFromImageTranspose,
|
|
"PtFromLatent": PtFromLatent,
|
|
"PtFromNumpy": PtFromNumpy,
|
|
"PtFull": PtFull,
|
|
"PtGather": PtGather,
|
|
"PtGe": PtGe,
|
|
"PtGt": PtGt,
|
|
"PtIndexSelect": PtIndexSelect,
|
|
"PtIntCreate": PtIntCreate,
|
|
"PtInterpolateByScaleFactor": PtInterpolateByScaleFactor,
|
|
"PtInterpolateToSize": PtInterpolateToSize,
|
|
"PtLe": PtLe,
|
|
"PtLinspace": PtLinspace,
|
|
"PtLoadModel": PtLoadModel,
|
|
"PtLog": PtLog,
|
|
"PtLogicalAnd": PtLogicalAnd,
|
|
"PtLogicalNot": PtLogicalNot,
|
|
"PtLogicalOr": PtLogicalOr,
|
|
"PtLogicalXor": PtLogicalXor,
|
|
"PtLt": PtLt,
|
|
"PtMaskedSelect": PtMaskedSelect,
|
|
"PtMatMul": PtMatMul,
|
|
"PtMax": PtMax,
|
|
"PtMean": PtMean,
|
|
"PtMedian": PtMedian,
|
|
"PtMin": PtMin,
|
|
"PtMm": PtMm,
|
|
"PtMul": PtMul,
|
|
"PtNe": PtNe,
|
|
"PtNeg": PtNeg,
|
|
"PtOnes": PtOnes,
|
|
"PtPad": PtPad,
|
|
"PtPermute": PtPermute,
|
|
"PtPow": PtPow,
|
|
"PtPredictClassificationModel": PtPredictClassificationModel,
|
|
"PtProd": PtProd,
|
|
"PtRand": PtRand,
|
|
"PtRandInt": PtRandInt,
|
|
"PtRandn": PtRandn,
|
|
"PtRemainder": PtRemainder,
|
|
"PtReshape": PtReshape,
|
|
"PtSaveModel": PtSaveModel,
|
|
"PtScatter": PtScatter,
|
|
"PtShowSize": PtShowSize,
|
|
"PtShowText": PtShowText,
|
|
"PtSin": PtSin,
|
|
"PtSinh": PtSinh,
|
|
"PtSize": PtSize,
|
|
"PtSizeCreate": PtSizeCreate,
|
|
"PtSizeToNumpy": PtSizeToNumpy,
|
|
"PtSizeToString": PtSizeToString,
|
|
"PtSqrt": PtSqrt,
|
|
"PtSqueeze": PtSqueeze,
|
|
"PtStd": PtStd,
|
|
"PtSub": PtSub,
|
|
"PtSum": PtSum,
|
|
"PtTan": PtTan,
|
|
"PtTanh": PtTanh,
|
|
"PtToImage": PtToImage,
|
|
"PtToImageTranspose": PtToImageTranspose,
|
|
"PtToLatent": PtToLatent,
|
|
"PtToNumpy": PtToNumpy,
|
|
"PtToRgbTensors": PtToRgbTensors,
|
|
"PtTrainClassificationModel": PtTrainClassificationModel,
|
|
"PtUnsqueeze": PtUnsqueeze,
|
|
"PtVar": PtVar,
|
|
"PtView": PtView,
|
|
"PtWhere": PtWhere,
|
|
"PtZeros": PtZeros,
|
|
"PtnConvModel": PtnConvModel,
|
|
"PtnLinearModel": PtnLinearModel,
|
|
"PtnResnetModel": PtnResnetModel,
|
|
"PtoAdam": PtoAdam,
|
|
"PtvDataset": PtvDataset,
|
|
"PtvDatasetLen": PtvDatasetLen,
|
|
"PtvDatasetLoader": PtvDatasetLoader,
|
|
"PtvImageFolderDataset": PtvImageFolderDataset,
|
|
"PtvTransformsResize": PtvTransformsResize,
|
|
"PtvTransformsToTensor": PtvTransformsToTensor,
|
|
}
|
|
|
|
|
|
"""
|
|
NODE_DISPLAY_NAME_MAPPINGS (Dict[str, str]):
|
|
A dictionary mapping node names to user-friendly display names.
|
|
"""
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS: Dict[str, str] = {
|
|
"PtAbs": "Pt Abs",
|
|
"PtAcos": "Pt Acos",
|
|
"PtAdd": "Pt Add",
|
|
"PtArange": "Pt Arange",
|
|
"PtArgmax": "Pt Argmax",
|
|
"PtArgmin": "Pt Argmin",
|
|
"PtAsin": "Pt Asin",
|
|
"PtAtan": "Pt Atan",
|
|
"PtBitwiseAnd": "Pt Bitwise And",
|
|
"PtBitwiseLeftShift": "Pt Bitwise Left Shift",
|
|
"PtBitwiseNot": "Pt Bitwise Not",
|
|
"PtBitwiseOr": "Pt Bitwise Or",
|
|
"PtBitwiseRightShift": "Pt Bitwise Right Shift",
|
|
"PtBitwiseXor": "Pt Bitwise Xor",
|
|
"PtBmm": "Pt Bmm",
|
|
"PtBoolCreate": "Pt Bool Create",
|
|
"PtCos": "Pt Cos",
|
|
"PtCosh": "Pt Cosh",
|
|
"PtCrop": "Pt Crop",
|
|
"PtDataLoader": "Pt Data Loader",
|
|
"PtDiv": "Pt Div",
|
|
"PtEinsum": "Pt Einsum",
|
|
"PtEq": "Pt Eq",
|
|
"PtEvaluateClassificationModel": "Pt Evaluate Classification Model",
|
|
"PtExp": "Pt Exp",
|
|
"PtFlatten": "Pt Flatten",
|
|
"PtFloatCreate": "Pt Float Create",
|
|
"PtFloorDiv": "Pt Floor Div",
|
|
"PtFromImage": "Pt From Image",
|
|
"PtFromImageTranspose": "Pt From Image Transpose",
|
|
"PtFromLatent": "Pt From Latent",
|
|
"PtFromNumpy": "Pt From Numpy",
|
|
"PtFull": "Pt Full",
|
|
"PtGather": "Pt Gather",
|
|
"PtGe": "Pt Ge",
|
|
"PtGt": "Pt Gt",
|
|
"PtIndexSelect": "Pt Index Select",
|
|
"PtIntCreate": "Pt Int Create",
|
|
"PtInterpolateByScaleFactor": "Pt Interpolate By Scale Factor",
|
|
"PtInterpolateToSize": "Pt Interpolate To Size",
|
|
"PtLe": "Pt Le",
|
|
"PtLinspace": "Pt Linspace",
|
|
"PtLoadModel": "Pt Load Model",
|
|
"PtLog": "Pt Log",
|
|
"PtLogicalAnd": "Pt Logical And",
|
|
"PtLogicalNot": "Pt Logical Not",
|
|
"PtLogicalOr": "Pt Logical Or",
|
|
"PtLogicalXor": "Pt Logical Xor",
|
|
"PtLt": "Pt Lt",
|
|
"PtMaskedSelect": "Pt Masked Select",
|
|
"PtMatMul": "Pt Mat Mul",
|
|
"PtMax": "Pt Max",
|
|
"PtMean": "Pt Mean",
|
|
"PtMedian": "Pt Median",
|
|
"PtMin": "Pt Min",
|
|
"PtMm": "Pt Mm",
|
|
"PtMul": "Pt Mul",
|
|
"PtNe": "Pt Ne",
|
|
"PtNeg": "Pt Neg",
|
|
"PtOnes": "Pt Ones",
|
|
"PtPad": "Pt Pad",
|
|
"PtPermute": "Pt Permute",
|
|
"PtPow": "Pt Pow",
|
|
"PtPredictClassificationModel": "Pt Predict Classification Model",
|
|
"PtProd": "Pt Prod",
|
|
"PtRand": "Pt Rand",
|
|
"PtRandInt": "Pt Rand Int",
|
|
"PtRandn": "Pt Randn",
|
|
"PtRemainder": "Pt Remainder",
|
|
"PtReshape": "Pt Reshape",
|
|
"PtSaveModel": "Pt Save Model",
|
|
"PtScatter": "Pt Scatter",
|
|
"PtShowSize": "Pt Show Size",
|
|
"PtShowText": "Pt Show Text",
|
|
"PtSin": "Pt Sin",
|
|
"PtSinh": "Pt Sinh",
|
|
"PtSize": "Pt Size",
|
|
"PtSizeCreate": "Pt Size Create",
|
|
"PtSizeToNumpy": "Pt Size To Numpy",
|
|
"PtSizeToString": "Pt Size To String",
|
|
"PtSqrt": "Pt Sqrt",
|
|
"PtSqueeze": "Pt Squeeze",
|
|
"PtStd": "Pt Std",
|
|
"PtSub": "Pt Sub",
|
|
"PtSum": "Pt Sum",
|
|
"PtTan": "Pt Tan",
|
|
"PtTanh": "Pt Tanh",
|
|
"PtToImage": "Pt To Image",
|
|
"PtToImageTranspose": "Pt To Image Transpose",
|
|
"PtToLatent": "Pt To Latent",
|
|
"PtToNumpy": "Pt To Numpy",
|
|
"PtToRgbTensors": "Pt To Rgb Tensors",
|
|
"PtTrainClassificationModel": "Pt Train Classification Model",
|
|
"PtUnsqueeze": "Pt Unsqueeze",
|
|
"PtVar": "Pt Var",
|
|
"PtView": "Pt View",
|
|
"PtWhere": "Pt Where",
|
|
"PtZeros": "Pt Zeros",
|
|
"PtnConvModel": "Ptn Conv Model",
|
|
"PtnLinearModel": "Ptn Linear Model",
|
|
"PtnResnetModel": "Ptn Resnet Model",
|
|
"PtoAdam": "Pto Adam",
|
|
"PtvDataset": "Ptv Dataset",
|
|
"PtvDatasetLen": "Ptv Dataset Len",
|
|
"PtvDatasetLoader": "Ptv Dataset Loader",
|
|
"PtvImageFolderDataset": "Ptv Image Folder Dataset",
|
|
"PtvTransformsResize": "Ptv Transforms Resize",
|
|
"PtvTransformsToTensor": "Ptv Transforms To Tensor",
|
|
}
|
|
|
|
|
|
"""
|
|
Below two lines were taken from:
|
|
https://github.com/pythongosssss/ComfyUI-Custom-Scripts/blob/main/__init__.py
|
|
See credit/credit.md for the full license.
|
|
"""
|
|
|
|
WEB_DIRECTORY = "./web"
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]
|