diff --git a/modules/audioInference.py b/modules/audioInference.py index a3c89eb..a12e396 100644 --- a/modules/audioInference.py +++ b/modules/audioInference.py @@ -51,22 +51,16 @@ class RunwareAudioInference: "default": False, "tooltip": "Enable/disable sampleRate parameter in API request" }), - "sampleRate": ("INT", { + "sampleRate": ([8000, 16000, 22050, 24000, 32000, 44100], { "default": 32000, - "min": 8000, - "max": 44100, - "step": 1, "tooltip": "Audio sample rate in Hz. Supported: 8000, 16000, 22050, 24000, 32000, 44100." }), "useBitrate": ("BOOLEAN", { "default": False, "tooltip": "Enable/disable bitrate parameter in API request (MP3 only)" }), - "bitrate": ("INT", { + "bitrate": ([32, 64, 128, 256], { "default": 128, - "min": 32, - "max": 256, - "step": 1, "tooltip": "Audio bitrate in kbps. Allowed: 32, 64, 128, 256. Only applies to MP3 format." }), "useChannels": ("BOOLEAN", { @@ -270,7 +264,7 @@ class RunwareAudioInference: "model": kwargs.get("model", ""), "duration": kwargs.get("duration", 30), "useDuration": kwargs.get("useDuration", True), - "sampleRate": kwargs.get("sampleRate", 32000), + "sampleRate": int(kwargs.get("sampleRate", 32000)), "useSampleRate": kwargs.get("useSampleRate", False), "bitrate": kwargs.get("bitrate", 128), "useBitrate": kwargs.get("useBitrate", False), @@ -326,9 +320,11 @@ class RunwareAudioInference: # Build audioSettings conditionally based on use flags audioSettings = {} if params["useSampleRate"]: - audioSettings["sampleRate"] = params["sampleRate"] + audioSettings["sampleRate"] = int(params["sampleRate"]) if params["useBitrate"]: - audioSettings["bitrate"] = params["bitrate"] + _br = int(params["bitrate"]) + _allowed_bitrates = (32, 64, 128, 256) + audioSettings["bitrate"] = _br if _br in _allowed_bitrates else min(_allowed_bitrates, key=lambda x: abs(x - _br)) if params["useChannels"]: audioSettings["channels"] = params["channels"] diff --git a/modules/providerSettings/briaImageProviderSettings.py b/modules/providerSettings/briaImageProviderSettings.py index 6f2b5a0..1d9797b 100644 --- a/modules/providerSettings/briaImageProviderSettings.py +++ b/modules/providerSettings/briaImageProviderSettings.py @@ -325,11 +325,11 @@ class RunwareBriaProviderSettings: if useColor: settings["color"] = color - # Add lightDirection only if useLightDirection is enabled (maps to /relight; API may expect light_direction) + # Add lightDirection only if useLightDirection is enabled (maps to /relight if useLightDirection: settings["lightDirection"] = lightDirection - # Add lightType only if useLightType is enabled (maps to /relight; API may expect light_type) + # Add lightType only if useLightType is enabled (maps to /relight) if useLightType: settings["lightType"] = lightType