89 lines
3.3 KiB
Python
89 lines
3.3 KiB
Python
"""
|
|
Runware 3D Inference Settings Tex Slat Node
|
|
Outputs texSlat config: guidanceStrength, guidanceRescale, steps, rescaleT
|
|
"""
|
|
|
|
from typing import Dict, Any
|
|
|
|
|
|
class Runware3DInferenceSettingsTexSlat:
|
|
"""Runware 3D Inference Settings Tex Slat Node"""
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {},
|
|
"optional": {
|
|
"useGuidanceStrength": ("BOOLEAN", {
|
|
"tooltip": "Enable to include guidanceStrength in texSlat",
|
|
"default": False,
|
|
}),
|
|
"guidanceStrength": ("FLOAT", {
|
|
"tooltip": "Guidance strength for tex slat (1.0-10.0). Only used when enabled.",
|
|
"default": 1.0,
|
|
"min": 1.0,
|
|
"max": 10.0,
|
|
"step": 0.1,
|
|
}),
|
|
"useGuidanceRescale": ("BOOLEAN", {
|
|
"tooltip": "Enable to include guidanceRescale in texSlat",
|
|
"default": False,
|
|
}),
|
|
"guidanceRescale": ("FLOAT", {
|
|
"tooltip": "Guidance rescale for tex slat. Only used when enabled.",
|
|
"default": 0.0,
|
|
"min": 0.0,
|
|
"max": 1.0,
|
|
"step": 0.05,
|
|
}),
|
|
"useSteps": ("BOOLEAN", {
|
|
"tooltip": "Enable to include steps in texSlat",
|
|
"default": False,
|
|
}),
|
|
"steps": ("INT", {
|
|
"tooltip": "Number of steps for tex slat (1-50). Only used when enabled.",
|
|
"default": 12,
|
|
"min": 1,
|
|
"max": 50,
|
|
"step": 1,
|
|
}),
|
|
"useRescaleT": ("BOOLEAN", {
|
|
"tooltip": "Enable to include rescaleT in texSlat",
|
|
"default": False,
|
|
}),
|
|
"rescaleT": ("FLOAT", {
|
|
"tooltip": "Rescale T for tex slat (1.0-6.0). Only used when enabled.",
|
|
"default": 3.0,
|
|
"min": 1.0,
|
|
"max": 6.0,
|
|
"step": 0.1,
|
|
}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("RUNWARE3DINFERENCESETTINGSLAT",)
|
|
RETURN_NAMES = ("Tex Slat",)
|
|
FUNCTION = "create"
|
|
CATEGORY = "Runware"
|
|
DESCRIPTION = "Configure tex slat 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", 1.0))
|
|
if kwargs.get("useGuidanceRescale", False):
|
|
out["guidanceRescale"] = float(kwargs.get("guidanceRescale", 0.0))
|
|
if kwargs.get("useSteps", False):
|
|
out["steps"] = int(kwargs.get("steps", 12))
|
|
if kwargs.get("useRescaleT", False):
|
|
out["rescaleT"] = float(kwargs.get("rescaleT", 3.0))
|
|
return (out,)
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"Runware3DInferenceSettingsTexSlat": Runware3DInferenceSettingsTexSlat,
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"Runware3DInferenceSettingsTexSlat": "Runware 3D Inference Settings Tex Slat",
|
|
}
|