modify the code

siliconflow · Dec 11, 2024 · 883608a · 883608a
1 parent 0540ef9
commit 883608a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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"