feat: add LTX-2 retake model

modules/providerSettings/lightricksProviderSettings.py

@@ -13,10 +13,49 @@ class RunwareLightricksProviderSettings:
         return {
             "required": {},
             "optional": {
+                
+                "useGenerateAudio": ("BOOLEAN", {
+                    "tooltip": "Enable to override the audio generation flag. Disable to keep the default behavior.",
+                    "default": False,
+                    "label_on": "Enabled",
+                    "label_off": "Disabled",
+                }),
                 "generateAudio": ("BOOLEAN", {
                     "tooltip": "Enable to generate audio for the video. Disable to generate video without audio. Default: false.",
                     "default": False,
                 }),
+                "useStartTime": ("BOOLEAN", {
+                    "tooltip": "Enable to include a custom start time for the replacement segment.",
+                    "default": False,
+                    "label_on": "Enabled",
+                    "label_off": "Disabled",
+                }),
+                "startTime": ("INT", {
+                    "tooltip": "Start time (seconds) where the replacement should begin.",
+                    "default": 0,
+                    "min": 0,
+                }),
+                "useDuration": ("BOOLEAN", {
+                    "tooltip": "Enable to include a custom duration for the replacement segment.",
+                    "default": False,
+                    "label_on": "Enabled",
+                    "label_off": "Disabled",
+                }),
+                "duration": ("INT", {
+                    "tooltip": "Duration (seconds) for the replacement segment.",
+                    "default": 5,
+                    "min": 1,
+                }),
+                "useMode": ("BOOLEAN", {
+                    "tooltip": "Enable to select what parts of the video should be replaced.",
+                    "default": False,
+                    "label_on": "Enabled",
+                    "label_off": "Default",
+                }),
+                "mode": (["replace_audio", "replace_video", "replace_audio_and_video"], {
+                    "tooltip": "Choose what to replace in the video task.",
+                    "default": "replace_audio",
+                }),
             }
         }
     
@@ -30,12 +69,25 @@ class RunwareLightricksProviderSettings:
         
         # Get value parameters
         generateAudio = kwargs.get("generateAudio", False)
+        useStartTime = kwargs.get("useStartTime", False)
+        useDuration = kwargs.get("useDuration", False)
+        useMode = kwargs.get("useMode", False)
+        useGenerateAudio = kwargs.get("useGenerateAudio", False)
+        startTime = kwargs.get("startTime", 0)
+        duration = kwargs.get("duration", 5)
+        mode = kwargs.get("mode", "replace_audio")
         
         # Build settings dictionary
         # Always include generateAudio to ensure it's explicitly set
-        settings = {
-            "generateAudio": generateAudio
-        }
+        settings = {}
+        if useGenerateAudio:
+            settings["generateAudio"] = generateAudio
+        if useStartTime:
+            settings["startTime"] = startTime
+        if useDuration:
+            settings["duration"] = duration
+        if useMode:
+            settings["mode"] = mode
         
         return (settings,)
 

modules/videoModelSearch.py

@@ -62,6 +62,7 @@ class videoModelSearch:
         "Lightricks": [
             "lightricks:2@0 (LTX Fast)",
             "lightricks:2@1 (LTX Pro)",
+            "lightricks:3@1 (LTX-2 Retake)",
         ],
         "Ovi": [
             "runware:190@1 (Ovi)",
@@ -138,6 +139,7 @@ class videoModelSearch:
         # Lightricks Models
         "lightricks:2@0": {"width": 1920, "height": 1080},
         "lightricks:2@1": {"width": 1920, "height": 1080},
+        "lightricks:3@1": {"width": 0, "height": 0},
         
         # Ovi Models
         "runware:190@1": {"width": 0, "height": 0},