add trellis support

2026-02-10 00:19:10 -05:00
parent 668d66b70e
commit 63bc583059
12 changed files with 510 additions and 1 deletions
@@ -0,0 +1,88 @@
+"""
+Runware 3D Inference Settings Sparse Structure Node
+Outputs sparseStructure config: guidanceStrength, guidanceRescale, steps, rescaleT
+"""
+
+from typing import Dict, Any
+
+
+class Runware3DInferenceSettingsSparseStructure:
+    """Runware 3D Inference Settings Sparse Structure Node"""
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {},
+            "optional": {
+                "useGuidanceStrength": ("BOOLEAN", {
+                    "tooltip": "Enable to include guidanceStrength in sparseStructure",
+                    "default": False,
+                }),
+                "guidanceStrength": ("FLOAT", {
+                    "tooltip": "Guidance strength for sparse structure (1.0-10.0). Only used when enabled.",
+                    "default": 7.5,
+                    "min": 1.0,
+                    "max": 10.0,
+                    "step": 0.1,
+                }),
+                "useGuidanceRescale": ("BOOLEAN", {
+                    "tooltip": "Enable to include guidanceRescale in sparseStructure",
+                    "default": False,
+                }),
+                "guidanceRescale": ("FLOAT", {
+                    "tooltip": "Guidance rescale for sparse structure. Only used when enabled.",
+                    "default": 0.7,
+                    "min": 0.0,
+                    "max": 1.0,
+                    "step": 0.05,
+                }),
+                "useSteps": ("BOOLEAN", {
+                    "tooltip": "Enable to include steps in sparseStructure",
+                    "default": False,
+                }),
+                "steps": ("INT", {
+                    "tooltip": "Number of steps for sparse structure (1-50). Only used when enabled.",
+                    "default": 12,
+                    "min": 1,
+                    "max": 50,
+                    "step": 1,
+                }),
+                "useRescaleT": ("BOOLEAN", {
+                    "tooltip": "Enable to include rescaleT in sparseStructure",
+                    "default": False,
+                }),
+                "rescaleT": ("FLOAT", {
+                    "tooltip": "Rescale T for sparse structure (1.0-6.0). Only used when enabled.",
+                    "default": 5.0,
+                    "min": 1.0,
+                    "max": 6.0,
+                    "step": 0.1,
+                }),
+            }
+        }
+
+    RETURN_TYPES = ("RUNWARE3DINFERENCESETTINGSLAT",)
+    RETURN_NAMES = ("Sparse Structure",)
+    FUNCTION = "create"
+    CATEGORY = "Runware"
+    DESCRIPTION = "Configure sparse structure parameters for Runware 3D Inference settings."
+
+    def create(self, **kwargs) -> tuple[Dict[str, Any]]:
+        out: Dict[str, Any] = {}
+        if kwargs.get("useGuidanceStrength", False):
+            out["guidanceStrength"] = float(kwargs.get("guidanceStrength", 7.5))
+        if kwargs.get("useGuidanceRescale", False):
+            out["guidanceRescale"] = float(kwargs.get("guidanceRescale", 0.7))
+        if kwargs.get("useSteps", False):
+            out["steps"] = int(kwargs.get("steps", 12))
+        if kwargs.get("useRescaleT", False):
+            out["rescaleT"] = float(kwargs.get("rescaleT", 5.0))
+        return (out,)
+
+
+NODE_CLASS_MAPPINGS = {
+    "Runware3DInferenceSettingsSparseStructure": Runware3DInferenceSettingsSparseStructure,
+}
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "Runware3DInferenceSettingsSparseStructure": "Runware 3D Inference Settings Sparse Structure",
+}