diff --git a/bizyair_extras/nodes_comfyui_rf_inversion.py b/bizyair_extras/nodes_comfyui_rf_inversion.py index 88b9eaa0..ad1b4441 100644 --- a/bizyair_extras/nodes_comfyui_rf_inversion.py +++ b/bizyair_extras/nodes_comfyui_rf_inversion.py @@ -6,7 +6,7 @@ from bizyair import BizyAirBaseNode, data_types -class OutFluxModelSamplingPredNode(BizyAirBaseNode): +class OutFluxModelSamplingPred(BizyAirBaseNode): DESCRIPTION = "" RETURN_TYPES = (data_types.MODEL,) NODE_DISPLAY_NAME = "Outverse Flux Model Pred" @@ -48,7 +48,7 @@ def INPUT_TYPES(s): } -class InFluxModelSamplingPredNode(BizyAirBaseNode): +class InFluxModelSamplingPred(BizyAirBaseNode): RETURN_TYPES = (data_types.MODEL,) CATEGORY = "sampling/custom_sampling/samplers" NODE_DISPLAY_NAME = "Inverse Flux Model Pred" @@ -88,7 +88,7 @@ def INPUT_TYPES(s): } -class FluxForwardODESamplerNode(BizyAirBaseNode): +class FluxForwardODESampler(BizyAirBaseNode): RETURN_TYPES = ("SAMPLER",) # FUNCTION = "build" CATEGORY = "sampling/custom_sampling/samplers" @@ -109,7 +109,7 @@ def INPUT_TYPES(s): } -class FluxReverseODESamplerNode(BizyAirBaseNode): +class FluxReverseODESampler(BizyAirBaseNode): RETURN_TYPES = ("SAMPLER",) # FUNCTION = "build" CATEGORY = "sampling/custom_sampling/samplers"