Added new params for Dia 2 support

clientlibs/utils.js

@@ -1251,6 +1251,12 @@ function audioSettingsToggleHandler(settingsNode) {
 
     const useTemperatureWidget = settingsNode.widgets.find(w => w && w.name === "useTemperature");
     const temperatureWidget = settingsNode.widgets.find(w => w && w.name === "temperature");
+    const useAudioTemperatureWidget = settingsNode.widgets.find(w => w && w.name === "useAudioTemperature");
+    const audioTemperatureWidget = settingsNode.widgets.find(w => w && w.name === "audioTemperature");
+    const useTopKWidget = settingsNode.widgets.find(w => w && w.name === "useTopK");
+    const topKWidget = settingsNode.widgets.find(w => w && w.name === "topK");
+    const useIncludePrefixWidget = settingsNode.widgets.find(w => w && w.name === "useIncludePrefix");
+    const includePrefixWidget = settingsNode.widgets.find(w => w && w.name === "includePrefix");
     const useLyricsWidget = settingsNode.widgets.find(w => w && w.name === "useLyrics");
     const lyricsWidget = settingsNode.widgets.find(w => w && w.name === "lyrics");
     const useGuidanceTypeWidget = settingsNode.widgets.find(w => w && w.name === "useGuidanceType");
@@ -1299,6 +1305,9 @@ function audioSettingsToggleHandler(settingsNode) {
     }
 
     if (useTemperatureWidget && temperatureWidget) toggleWidgetState(useTemperatureWidget, temperatureWidget, "temperature");
+    if (useAudioTemperatureWidget && audioTemperatureWidget) toggleWidgetState(useAudioTemperatureWidget, audioTemperatureWidget, "audioTemperature");
+    if (useTopKWidget && topKWidget) toggleWidgetState(useTopKWidget, topKWidget, "topK");
+    if (useIncludePrefixWidget && includePrefixWidget) toggleWidgetState(useIncludePrefixWidget, includePrefixWidget, "includePrefix");
     if (useLyricsWidget && lyricsWidget) toggleWidgetState(useLyricsWidget, lyricsWidget, "lyrics");
     if (useGuidanceTypeWidget && guidanceTypeWidget) toggleWidgetState(useGuidanceTypeWidget, guidanceTypeWidget, "guidanceType");
     if (useLanguageBoostWidget && languageBoostWidget) toggleWidgetState(useLanguageBoostWidget, languageBoostWidget, "languageBoost");
@@ -3040,6 +3049,7 @@ function audioModelSearchFilterHandler(audioModelSearchNode) {
         ],
         "Dia": [
             "runware:dia@v1.0 (Dia 1.6B)",
+            "runware:dia2@2b (Dia2 2B)",
         ],
         "xAI": [
             "xai:voice@tts (xAI TTS)",

modules/audioInference.py

@@ -136,7 +136,7 @@ class RunwareAudioInference:
                     "tooltip": "Custom inputs for audio generation (e.g., video URL for audio extraction)"
                 }),
                 "settings": ("RUNWAREAUDIOSETTINGS", {
-                    "tooltip": "Connect Runware Audio Inference Settings for lyrics, guidanceType, languageBoost, turbo, temperature, textNormalization, xVectorOnly, maxNewTokens, transcript, etc."
+                    "tooltip": "Connect Runware Audio Inference Settings for includePrefix, topK, temperature, audioTemperature, lyrics, guidanceType, languageBoost, turbo, textNormalization, xVectorOnly, maxNewTokens, transcript, etc."
                 }),
                 "speech": ("RUNWARESPEECH", {
                     "tooltip": "Connect Runware Audio Inference Speech for speech synthesis (e.g. Minimax)"

modules/audioModelSearch.py

@@ -26,6 +26,7 @@ class RunwareAudioModelSearch:
         ],
         "Dia": [
             "runware:dia@v1.0 (Dia 1.6B)",
+            "runware:dia2@2b (Dia2 2B)",
         ],
         "xAI": [
             "xai:voice@tts (xAI TTS)",

modules/audioSettings.py

@@ -28,12 +28,44 @@ class RunwareAudioSettings:
                     "default": False,
                 }),
                 "temperature": ("FLOAT", {
-                    "tooltip": "Temperature value for audio generation. Higher values increase diversity, lower values make output more deterministic. Only used when 'Use Temperature' is enabled.",
-                    "default": 1.0,
+                    "tooltip": "Sampling temperature for text token generation. Lower values produce more coherent, predictable speech content. Higher values increase variety but risk artifacts. Set 0 for greedy decoding. Only used when 'Use Temperature' is enabled.",
+                    "default": 0.6,
                     "min": 0.0,
-                    "max": 2.0,
+                    "max": 5.0,
                     "step": 0.1,
                 }),
+                "useAudioTemperature": ("BOOLEAN", {
+                    "tooltip": "Enable to include audioTemperature in audio generation settings",
+                    "default": False,
+                }),
+                "audioTemperature": ("FLOAT", {
+                    "tooltip": "Sampling temperature for audio token generation. Lower values produce cleaner, more stable audio. Higher values add expressiveness but may introduce noise. Set 0 for greedy decoding. Only used when 'Use Audio Temperature' is enabled.",
+                    "default": 0.8,
+                    "min": 0.0,
+                    "max": 5.0,
+                    "step": 0.1,
+                }),
+                "useTopK": ("BOOLEAN", {
+                    "tooltip": "Enable to include topK token sampling in audio generation settings",
+                    "default": False,
+                }),
+                "topK": ("INT", {
+                    "tooltip": "Top-K candidates for both text and audio token sampling. Limits sampling to the K most likely next tokens at each step. Only used when 'Use Top K' is enabled.",
+                    "default": 50,
+                    "min": 1,
+                    "max": 500,
+                    "step": 1,
+                }),
+                "useIncludePrefix": ("BOOLEAN", {
+                    "tooltip": "Enable to include includePrefix in audio generation settings",
+                    "default": False,
+                }),
+                "includePrefix": ("BOOLEAN", {
+                    "tooltip": "When voice cloning, keep prefix audio in output (true) or trim to new speech (false). Requires at least prefix_audio_speaker_1 input. Only used when 'Use Include Prefix' is enabled.",
+                    "default": False,
+                    "label_on": "true",
+                    "label_off": "false",
+                }),
                 "useLyrics": ("BOOLEAN", {
                     "tooltip": "Enable to include lyrics in audio generation settings",
                     "default": False,
@@ -198,6 +230,12 @@ class RunwareAudioSettings:
         """Create audio settings dict for API"""
         use_temperature = kwargs.get("useTemperature", False)
         temperature = kwargs.get("temperature", 1.0)
+        use_audio_temperature = kwargs.get("useAudioTemperature", False)
+        audio_temperature = kwargs.get("audioTemperature", 0.8)
+        use_top_k = kwargs.get("useTopK", False)
+        top_k = kwargs.get("topK", 50)
+        use_include_prefix = kwargs.get("useIncludePrefix", False)
+        include_prefix = kwargs.get("includePrefix", False)
         use_lyrics = kwargs.get("useLyrics", False)
         lyrics = kwargs.get("lyrics", "")
         use_guidance_type = kwargs.get("useGuidanceType", False)
@@ -233,6 +271,12 @@ class RunwareAudioSettings:
 
         if use_temperature:
             settings["temperature"] = float(temperature)
+        if use_audio_temperature:
+            settings["audioTemperature"] = float(audio_temperature)
+        if use_top_k:
+            settings["topK"] = int(top_k)
+        if use_include_prefix:
+            settings["includePrefix"] = bool(include_prefix)
 
         if use_lyrics and lyrics and lyrics.strip():
             settings["lyrics"] = lyrics.strip()