add rf inversion nodes

backend/bizyengine/bizyair_extras/__init__.py

@@ -15,6 +15,7 @@
 from .nodes_janus_pro import *
 from .nodes_kolors_mz import *
 from .nodes_model_advanced import *
+from .nodes_rf_inversion import *
 from .nodes_sd3 import *
 from .nodes_segment_anything import *
 from .nodes_testing_utils import *

backend/bizyengine/bizyair_extras/nodes_rf_inversion.py

@@ -0,0 +1,156 @@
+"""
+Reference: https://github.com/logtd/ComfyUI-Fluxtapoz
+"""
+
+# from bizyair import BizyAirBaseNode, data_types
+import folder_paths
+import nodes
+from bizyengine.core import BizyAirBaseNode, BizyAirNodeIO, create_node_data, data_types
+
+
+class BizyAir_OutFluxModelSamplingPred(BizyAirBaseNode):
+    DESCRIPTION = ""
+    RETURN_TYPES = (data_types.MODEL,)
+    NODE_DISPLAY_NAME = "☁️BizyAir Outverse Flux Model Pred"
+    CATEGORY = "☁️BizyAir/sampling/custom_sampling/samplers"
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "model": (data_types.MODEL,),
+                "max_shift": (
+                    "FLOAT",
+                    {"default": 1.15, "min": 0.0, "max": 100.0, "step": 0.01},
+                ),
+                "base_shift": (
+                    "FLOAT",
+                    {"default": 0.5, "min": 0.0, "max": 100.0, "step": 0.01},
+                ),
+                "width": (
+                    "INT",
+                    {
+                        "default": 1024,
+                        "min": 16,
+                        "max": nodes.MAX_RESOLUTION,
+                        "step": 8,
+                    },
+                ),
+                "height": (
+                    "INT",
+                    {
+                        "default": 1024,
+                        "min": 16,
+                        "max": nodes.MAX_RESOLUTION,
+                        "step": 8,
+                    },
+                ),
+                "reverse_ode": ("BOOLEAN", {"default": False}),
+            }
+        }
+
+
+class BizyAir_InFluxModelSamplingPred(BizyAirBaseNode):
+    RETURN_TYPES = (data_types.MODEL,)
+    CATEGORY = "☁️BizyAir/sampling/custom_sampling/samplers"
+    NODE_DISPLAY_NAME = "☁️BizyAir Inverse Flux Model Pred"
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "model": (data_types.MODEL,),
+                "max_shift": (
+                    "FLOAT",
+                    {"default": 1.15, "min": 0.0, "max": 100.0, "step": 0.01},
+                ),
+                "base_shift": (
+                    "FLOAT",
+                    {"default": 0.5, "min": 0.0, "max": 100.0, "step": 0.01},
+                ),
+                "width": (
+                    "INT",
+                    {
+                        "default": 1024,
+                        "min": 16,
+                        "max": nodes.MAX_RESOLUTION,
+                        "step": 8,
+                    },
+                ),
+                "height": (
+                    "INT",
+                    {
+                        "default": 1024,
+                        "min": 16,
+                        "max": nodes.MAX_RESOLUTION,
+                        "step": 8,
+                    },
+                ),
+            }
+        }
+
+
+class BizyAir_FluxForwardODESampler(BizyAirBaseNode):
+    RETURN_TYPES = ("SAMPLER",)
+    # FUNCTION = "build"
+    CATEGORY = "☁️BizyAir/sampling/custom_sampling/samplers"
+    NODE_DISPLAY_NAME = "☁️BizyAir Flux Forward ODE Sampler"
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "gamma": (
+                    "FLOAT",
+                    {"default": 0.5, "min": 0.0, "max": 100.0, "step": 0.01},
+                ),
+            },
+            "optional": {
+                "seed": ("INT", {"default": 0, "min": 0, "max": 0xFFFFFFFFFFFFFFFF}),
+            },
+        }
+
+
+class BizyAir_FluxReverseODESampler(BizyAirBaseNode):
+    RETURN_TYPES = ("SAMPLER",)
+    # FUNCTION = "build"
+    CATEGORY = "☁️BizyAir/sampling/custom_sampling/samplers"
+    NODE_DISPLAY_NAME = "☁️BizyAir Flux Reverse ODE Sampler"
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "model": (data_types.MODEL,),
+                "latent_image": ("LATENT",),
+                "eta": (
+                    "FLOAT",
+                    {"default": 0.8, "min": 0.0, "max": 100.0, "step": 0.01},
+                ),
+                "start_step": ("INT", {"default": 0, "min": 0, "max": 1000, "step": 1}),
+                "end_step": ("INT", {"default": 5, "min": 0, "max": 1000, "step": 1}),
+            },
+            "optional": {
+                "eta_trend": (["constant", "linear_increase", "linear_decrease"],)
+            },
+        }
+
+
+class BizyAir_FluxDeGuidance(BizyAirBaseNode):
+
+    RETURN_TYPES = (data_types.CONDITIONING,)
+
+    CATEGORY = "☁️BizyAir/fluxtapoz"
+    NODE_DISPLAY_NAME = "☁️BizyAir Flux DeGuidance"
+
+    @classmethod
+    def INPUT_TYPES(s):
+        return {
+            "required": {
+                "conditioning": (data_types.CONDITIONING,),
+                "guidance": (
+                    "FLOAT",
+                    {"default": 3.5, "min": -100.0, "max": 100.0, "step": 0.1},
+                ),
+            }
+        }