Added Support for Luma ray 3.2

2026-06-19 22:39:56 +05:30
parent 83113c70ef
commit c42795833e
11 changed files with 369 additions and 17 deletions
@@ -0,0 +1,137 @@
+"""
+Runware Video Inference Settings Edit
+Exposes settings.edit (autoControls, strength, controls) for video editing models.
+Connect the output to Runware Video Inference Settings → edit.
+"""
+
+from typing import Dict, Any
+
+_STRENGTH_PRESETS = [
+    "adhere_1", "adhere_2", "adhere_3",
+    "flex_1", "flex_2", "flex_3",
+    "reimagine_1", "reimagine_2", "reimagine_3",
+]
+
+
+class RunwareVideoInferenceSettingsControl:
+    """Optional settings.edit block for video inference (e.g. Luma Ray 3.2)."""
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {},
+            "optional": {
+                "useAutoControls": ("BOOLEAN", {
+                    "tooltip": "Enable to include autoControls in settings.edit.",
+                    "default": False,
+                }),
+                "autoControls": ("BOOLEAN", {
+                    "tooltip": "Derives the full conditioning schedule from the source video. Cannot be combined with settings.edit.strength or settings.edit.controls. Only used when 'Use Auto Controls' is enabled.",
+                    "default": False,
+                    "label_on": "Enabled",
+                    "label_off": "Disabled",
+                }),
+                "useStrength": ("BOOLEAN", {
+                    "tooltip": "Enable to include strength in settings.edit.",
+                    "default": False,
+                }),
+                "strength": (_STRENGTH_PRESETS, {
+                    "tooltip": "Preservation preset: adhere (stay close to source), flex (moderate changes), or reimagine (loose guidance). Cannot be combined with autoControls. Only used when 'Use Strength' is enabled.",
+                    "default": "flex_2",
+                }),
+                "useDepthBlur": ("BOOLEAN", {
+                    "tooltip": "Enable to include depthBlur in settings.edit.controls.",
+                    "default": False,
+                }),
+                "depthBlur": ("FLOAT", {
+                    "tooltip": "Scene-geometry depth blur (0–1). Higher values allow more freedom. Only used when 'Use Depth Blur' is enabled.",
+                    "default": 0.0,
+                    "min": 0.0,
+                    "max": 1.0,
+                    "step": 0.01,
+                }),
+                "useFace": ("BOOLEAN", {
+                    "tooltip": "Enable to include face in settings.edit.controls.",
+                    "default": False,
+                }),
+                "face": ("BOOLEAN", {
+                    "tooltip": "Enable face-identity conditioning. Only used when 'Use Face' is enabled.",
+                    "default": False,
+                    "label_on": "Enabled",
+                    "label_off": "Disabled",
+                }),
+                "useNormalsAugmentation": ("BOOLEAN", {
+                    "tooltip": "Enable to include normalsAugmentation in settings.edit.controls.",
+                    "default": False,
+                }),
+                "normalsAugmentation": ("FLOAT", {
+                    "tooltip": "Surface-geometry normals augmentation (0–1). Higher values allow more reinterpretation. Only used when 'Use Normals Augmentation' is enabled.",
+                    "default": 0.0,
+                    "min": 0.0,
+                    "max": 1.0,
+                    "step": 0.01,
+                }),
+                "usePoseStrength": ("BOOLEAN", {
+                    "tooltip": "Enable to include poseStrength in settings.edit.controls.",
+                    "default": False,
+                }),
+                "poseStrength": (["precise", "coarse"], {
+                    "tooltip": "Skeleton conditioning strength. Only used when 'Use Pose Strength' is enabled.",
+                    "default": "precise",
+                }),
+                "useTrajectorySparsity": ("BOOLEAN", {
+                    "tooltip": "Enable to include trajectorySparsity in settings.edit.controls.",
+                    "default": False,
+                }),
+                "trajectorySparsity": ("FLOAT", {
+                    "tooltip": "Motion-anchoring sparsity (0–1). Higher values use fewer anchors. Only used when 'Use Trajectory Sparsity' is enabled.",
+                    "default": 0.0,
+                    "min": 0.0,
+                    "max": 1.0,
+                    "step": 0.01,
+                }),
+            },
+        }
+
+    RETURN_TYPES = ("RUNWAREVIDEOINFERENCESETTINGSEDIT",)
+    RETURN_NAMES = ("edit",)
+    FUNCTION = "create"
+    CATEGORY = "Runware"
+    DESCRIPTION = (
+        "Configure settings.edit (autoControls, strength, controls) for Runware Video Inference. "
+        "Connect to Runware Video Inference Settings → edit."
+    )
+
+    def create(self, **kwargs) -> tuple[Dict[str, Any]]:
+        edit: Dict[str, Any] = {}
+
+        if kwargs.get("useAutoControls", False):
+            edit["autoControls"] = bool(kwargs.get("autoControls", False))
+        if kwargs.get("useStrength", False):
+            edit["strength"] = str(kwargs.get("strength", "flex_2"))
+
+        controls: Dict[str, Any] = {}
+        if kwargs.get("useDepthBlur", False):
+            controls["depthBlur"] = float(kwargs.get("depthBlur", 0.0))
+        if kwargs.get("useFace", False):
+            controls["face"] = bool(kwargs.get("face", False))
+        if kwargs.get("useNormalsAugmentation", False):
+            controls["normalsAugmentation"] = float(kwargs.get("normalsAugmentation", 0.0))
+        if kwargs.get("usePoseStrength", False):
+            controls["poseStrength"] = str(kwargs.get("poseStrength", "precise"))
+        if kwargs.get("useTrajectorySparsity", False):
+            controls["trajectorySparsity"] = float(kwargs.get("trajectorySparsity", 0.0))
+
+        if len(controls) > 0:
+            edit["controls"] = controls
+
+        return (edit,)
+
+
+NODE_CLASS_MAPPINGS = {
+    "RunwareVideoInferenceSettingsControl": RunwareVideoInferenceSettingsControl,
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "RunwareVideoInferenceSettingsControl": "Runware Video Inference Settings Edit",
+}