# 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_apply_function import PtApplyFunction
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_data_loader_from_tensors import PtDataLoaderFromTensors
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_predict_regression_model import PtPredictRegressionModel
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_bfloat16 import PtToBfloat16
from .modules.pytorch_wrapper.pt_to_float16 import PtToFloat16
from .modules.pytorch_wrapper.pt_to_float32 import PtToFloat32
from .modules.pytorch_wrapper.pt_to_float64 import PtToFloat64
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_int16 import PtToInt16
from .modules.pytorch_wrapper.pt_to_int32 import PtToInt32
from .modules.pytorch_wrapper.pt_to_int64 import PtToInt64
from .modules.pytorch_wrapper.pt_to_int8 import PtToInt8
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_to_uint8 import PtToUint8
from .modules.pytorch_wrapper.pt_train_classification_model import PtTrainClassificationModel
from .modules.pytorch_wrapper.pt_train_classification_model_lr import PtTrainClassificationModelLr
from .modules.pytorch_wrapper.pt_train_regression_model import PtTrainRegressionModel
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.ptd_bernoulli import PtdBernoulli
from .modules.pytorch_wrapper.ptd_beta import PtdBeta
from .modules.pytorch_wrapper.ptd_binomial import PtdBinomial
from .modules.pytorch_wrapper.ptd_categorical import PtdCategorical
from .modules.pytorch_wrapper.ptd_chi2 import PtdChi2
from .modules.pytorch_wrapper.ptd_exponential import PtdExponential
from .modules.pytorch_wrapper.ptd_gamma import PtdGamma
from .modules.pytorch_wrapper.ptd_normal import PtdNormal
from .modules.pytorch_wrapper.ptd_poisson import PtdPoisson
from .modules.pytorch_wrapper.ptd_student_t import PtdStudentT
from .modules.pytorch_wrapper.ptd_uniform import PtdUniform
from .modules.pytorch_wrapper.ptdm_cdf import PtdmCdf
from .modules.pytorch_wrapper.ptdm_cdf_tensor import PtdmCdfTensor
from .modules.pytorch_wrapper.ptdm_log_prob import PtdmLogProb
from .modules.pytorch_wrapper.ptdm_log_prob_tensor import PtdmLogProbTensor
from .modules.pytorch_wrapper.ptdm_pdf import PtdmPdf
from .modules.pytorch_wrapper.ptdm_pdf_tensor import PtdmPdfTensor
from .modules.pytorch_wrapper.ptdm_pmf import PtdmPmf
from .modules.pytorch_wrapper.ptdm_pmf_tensor import PtdmPmfTensor
from .modules.pytorch_wrapper.ptdm_sample import PtdmSample
from .modules.pytorch_wrapper.ptf_gelu import PtfGELU
from .modules.pytorch_wrapper.ptf_leaky_relu import PtfLeakyReLU
from .modules.pytorch_wrapper.ptf_log_softmax import PtfLogSoftmax
from .modules.pytorch_wrapper.ptf_relu import PtfReLU
from .modules.pytorch_wrapper.ptf_sigmoid import PtfSigmoid
from .modules.pytorch_wrapper.ptf_silu import PtfSiLU
from .modules.pytorch_wrapper.ptf_softmax import PtfSoftmax
from .modules.pytorch_wrapper.ptf_softplus import PtfSoftplus
from .modules.pytorch_wrapper.ptf_tanh import PtfTanh
from .modules.pytorch_wrapper.ptn_avg_pool2d import PtnAvgPool2d
from .modules.pytorch_wrapper.ptn_chained_model import PtnChainedModel
from .modules.pytorch_wrapper.ptn_conv2d import PtnConv2d
from .modules.pytorch_wrapper.ptn_conv_model import PtnConvModel
from .modules.pytorch_wrapper.ptn_linear import PtnLinear
from .modules.pytorch_wrapper.ptn_linear_model import PtnLinearModel
from .modules.pytorch_wrapper.ptn_max_pool2d import PtnMaxPool2d
from .modules.pytorch_wrapper.ptn_pre_add_channel_axis import PtnPreAddChannelAxis
from .modules.pytorch_wrapper.ptn_pre_flatten import PtnPreFlatten
from .modules.pytorch_wrapper.ptn_resnet_model import PtnResnetModel
from .modules.pytorch_wrapper.pto_adam import PtoAdam
from .modules.pytorch_wrapper.pto_adamw import PtoAdamW
from .modules.pytorch_wrapper.pto_lr_scheduler_cosine_annealing import PtoLrSchedulerCosineAnnealing
from .modules.pytorch_wrapper.pto_lr_scheduler_reduce_on_plateau import PtoLrSchedulerReduceOnPlateau
from .modules.pytorch_wrapper.pto_lr_scheduler_step import PtoLrSchedulerStep
from .modules.pytorch_wrapper.pto_sgd import PtoSGD
from .modules.pytorch_wrapper.pto_simple import PtoSimple
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_data_augment import PtvTransformsDataAugment
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,
    "PtApplyFunction": PtApplyFunction,
    "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,
    "PtDataLoaderFromTensors": PtDataLoaderFromTensors,
    "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,
    "PtPredictRegressionModel": PtPredictRegressionModel,
    "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,
    "PtToBfloat16": PtToBfloat16,
    "PtToFloat16": PtToFloat16,
    "PtToFloat32": PtToFloat32,
    "PtToFloat64": PtToFloat64,
    "PtToImage": PtToImage,
    "PtToImageTranspose": PtToImageTranspose,
    "PtToInt16": PtToInt16,
    "PtToInt32": PtToInt32,
    "PtToInt64": PtToInt64,
    "PtToInt8": PtToInt8,
    "PtToLatent": PtToLatent,
    "PtToNumpy": PtToNumpy,
    "PtToRgbTensors": PtToRgbTensors,
    "PtToUint8": PtToUint8,
    "PtTrainClassificationModel": PtTrainClassificationModel,
    "PtTrainClassificationModelLr": PtTrainClassificationModelLr,
    "PtTrainRegressionModel": PtTrainRegressionModel,
    "PtUnsqueeze": PtUnsqueeze,
    "PtVar": PtVar,
    "PtView": PtView,
    "PtWhere": PtWhere,
    "PtZeros": PtZeros,
    "PtdBernoulli": PtdBernoulli,
    "PtdBeta": PtdBeta,
    "PtdBinomial": PtdBinomial,
    "PtdCategorical": PtdCategorical,
    "PtdChi2": PtdChi2,
    "PtdExponential": PtdExponential,
    "PtdGamma": PtdGamma,
    "PtdNormal": PtdNormal,
    "PtdPoisson": PtdPoisson,
    "PtdStudentT": PtdStudentT,
    "PtdUniform": PtdUniform,
    "PtdmCdf": PtdmCdf,
    "PtdmCdfTensor": PtdmCdfTensor,
    "PtdmLogProb": PtdmLogProb,
    "PtdmLogProbTensor": PtdmLogProbTensor,
    "PtdmPdf": PtdmPdf,
    "PtdmPdfTensor": PtdmPdfTensor,
    "PtdmPmf": PtdmPmf,
    "PtdmPmfTensor": PtdmPmfTensor,
    "PtdmSample": PtdmSample,
    "PtfGELU": PtfGELU,
    "PtfLeakyReLU": PtfLeakyReLU,
    "PtfLogSoftmax": PtfLogSoftmax,
    "PtfReLU": PtfReLU,
    "PtfSiLU": PtfSiLU,
    "PtfSigmoid": PtfSigmoid,
    "PtfSoftmax": PtfSoftmax,
    "PtfSoftplus": PtfSoftplus,
    "PtfTanh": PtfTanh,
    "PtnAvgPool2d": PtnAvgPool2d,
    "PtnChainedModel": PtnChainedModel,
    "PtnConv2d": PtnConv2d,
    "PtnConvModel": PtnConvModel,
    "PtnLinear": PtnLinear,
    "PtnLinearModel": PtnLinearModel,
    "PtnMaxPool2d": PtnMaxPool2d,
    "PtnPreAddChannelAxis": PtnPreAddChannelAxis,
    "PtnPreFlatten": PtnPreFlatten,
    "PtnResnetModel": PtnResnetModel,
    "PtoAdam": PtoAdam,
    "PtoAdamW": PtoAdamW,
    "PtoLrSchedulerCosineAnnealing": PtoLrSchedulerCosineAnnealing,
    "PtoLrSchedulerReduceOnPlateau": PtoLrSchedulerReduceOnPlateau,
    "PtoLrSchedulerStep": PtoLrSchedulerStep,
    "PtoSGD": PtoSGD,
    "PtoSimple": PtoSimple,
    "PtvDataset": PtvDataset,
    "PtvDatasetLen": PtvDatasetLen,
    "PtvDatasetLoader": PtvDatasetLoader,
    "PtvImageFolderDataset": PtvImageFolderDataset,
    "PtvTransformsDataAugment": PtvTransformsDataAugment,
    "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",
    "PtApplyFunction": "Pt Apply Function",
    "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",
    "PtDataLoaderFromTensors": "Pt Data Loader From Tensors",
    "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",
    "PtPredictRegressionModel": "Pt Predict Regression 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",
    "PtToBfloat16": "Pt To Bfloat16",
    "PtToFloat16": "Pt To Float16",
    "PtToFloat32": "Pt To Float32",
    "PtToFloat64": "Pt To Float64",
    "PtToImage": "Pt To Image",
    "PtToImageTranspose": "Pt To Image Transpose",
    "PtToInt16": "Pt To Int16",
    "PtToInt32": "Pt To Int32",
    "PtToInt64": "Pt To Int64",
    "PtToInt8": "Pt To Int8",
    "PtToLatent": "Pt To Latent",
    "PtToNumpy": "Pt To Numpy",
    "PtToRgbTensors": "Pt To Rgb Tensors",
    "PtToUint8": "Pt To Uint8",
    "PtTrainClassificationModel": "Pt Train Classification Model",
    "PtTrainClassificationModelLr": "Pt Train Classification Model Lr",
    "PtTrainRegressionModel": "Pt Train Regression Model",
    "PtUnsqueeze": "Pt Unsqueeze",
    "PtVar": "Pt Var",
    "PtView": "Pt View",
    "PtWhere": "Pt Where",
    "PtZeros": "Pt Zeros",
    "PtdBernoulli": "Ptd Bernoulli",
    "PtdBeta": "Ptd Beta",
    "PtdBinomial": "Ptd Binomial",
    "PtdCategorical": "Ptd Categorical",
    "PtdChi2": "Ptd Chi2",
    "PtdExponential": "Ptd Exponential",
    "PtdGamma": "Ptd Gamma",
    "PtdNormal": "Ptd Normal",
    "PtdPoisson": "Ptd Poisson",
    "PtdStudentT": "Ptd Student T",
    "PtdUniform": "Ptd Uniform",
    "PtdmCdf": "Ptdm Cdf",
    "PtdmCdfTensor": "Ptdm Cdf Tensor",
    "PtdmLogProb": "Ptdm Log Prob",
    "PtdmLogProbTensor": "Ptdm Log Prob Tensor",
    "PtdmPdf": "Ptdm Pdf",
    "PtdmPdfTensor": "Ptdm Pdf Tensor",
    "PtdmPmf": "Ptdm Pmf",
    "PtdmPmfTensor": "Ptdm Pmf Tensor",
    "PtdmSample": "Ptdm Sample",
    "PtfGELU": "Ptf GELU",
    "PtfLeakyReLU": "Ptf Leaky ReLU",
    "PtfLogSoftmax": "Ptf Log Softmax",
    "PtfReLU": "Ptf ReLU",
    "PtfSiLU": "Ptf SiLU",
    "PtfSigmoid": "Ptf Sigmoid",
    "PtfSoftmax": "Ptf Softmax",
    "PtfSoftplus": "Ptf Softplus",
    "PtfTanh": "Ptf Tanh",
    "PtnAvgPool2d": "Ptn Avg Pool2d",
    "PtnChainedModel": "Ptn Chained Model",
    "PtnConv2d": "Ptn Conv2d",
    "PtnConvModel": "Ptn Conv Model",
    "PtnLinear": "Ptn Linear",
    "PtnLinearModel": "Ptn Linear Model",
    "PtnMaxPool2d": "Ptn Max Pool2d",
    "PtnPreAddChannelAxis": "Ptn Pre Add Channel Axis",
    "PtnPreFlatten": "Ptn Pre Flatten",
    "PtnResnetModel": "Ptn Resnet Model",
    "PtoAdam": "Pto Adam",
    "PtoAdamW": "Pto AdamW",
    "PtoLrSchedulerCosineAnnealing": "Pto Lr Scheduler Cosine Annealing",
    "PtoLrSchedulerReduceOnPlateau": "Pto Lr Scheduler Reduce On Plateau",
    "PtoLrSchedulerStep": "Pto Lr Scheduler Step",
    "PtoSGD": "Pto SGD",
    "PtoSimple": "Pto Simple",
    "PtvDataset": "Ptv Dataset",
    "PtvDatasetLen": "Ptv Dataset Len",
    "PtvDatasetLoader": "Ptv Dataset Loader",
    "PtvImageFolderDataset": "Ptv Image Folder Dataset",
    "PtvTransformsDataAugment": "Ptv Transforms Data Augment",
    "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"]