nodes/Runware-ComfyUI-Runware
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #44 from Runware/feature-Midjourneysupport/fix-videoUpscale

Browse Source 
add support for midjourney provider settings
This commit is contained in:
Sirshendu Ganguly
2025-11-14 18:44:41 -05:00
committed by GitHub
parent d89ac4ee8b 8c4cbd5ee6
commit efd7f0b0be
10 changed files with 464 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+2
View File
@@ -49,6 +49,7 @@ from .modules.providerSettings.minimaxProviderSettings import RunwareMiniMaxProv
from .modules.providerSettings.klingProviderSettings import RunwareKlingProviderSettings
from .modules.providerSettings.runwayProviderSettings import RunwareRunwayProviderSettings
from .modules.providerSettings.lumaProviderSettings import RunwareLumaProviderSettings
from .modules.providerSettings.midjourneyProviderSettings import RunwareMidjourneyProviderSettings
from .modules.vectorize import vectorize
from .modules.videoBgRemoval import videoBgRemoval
from .modules.videoUpscaler import videoUpscaler
@@ -119,6 +120,7 @@ NODE_CLASS_MAPPINGS = {
"Runware Luma Provider Settings": RunwareLumaProviderSettings,
"Runware KlingAI Provider Settings": RunwareKlingProviderSettings,
"Runware Runway Provider Settings": RunwareRunwayProviderSettings,
"Runware Midjourney Provider Settings": RunwareMidjourneyProviderSettings,
"Runware Vectorize": vectorize,
"Runware Video Background Removal": videoBgRemoval,
"Runware Video Upscaler": videoUpscaler,
clientlibs/main.js
+5 -1
View File
@@ -1,6 +1,6 @@
import { app } from "../../scripts/app.js";
import { api } from "../../scripts/api.js";
import { promptEnhanceHandler, syncDimensionsNodeHandler, searchNodeHandler, APIKeyHandler, captionNodeHandler, mediaUUIDHandler, videoTranscriptionHandler, handleCustomErrors, videoInferenceDimensionsHandler, videoModelSearchFilterHandler, audioModelSearchFilterHandler, useParameterToggleHandler, imageInferenceToggleHandler, upscalerToggleHandler, audioInferenceToggleHandler, acceleratorOptionsToggleHandler, bytedanceProviderSettingsToggleHandler, openaiProviderSettingsToggleHandler, klingProviderSettingsToggleHandler, lumaProviderSettingsToggleHandler } from "./utils.js";
import { promptEnhanceHandler, syncDimensionsNodeHandler, searchNodeHandler, APIKeyHandler, captionNodeHandler, mediaUUIDHandler, videoTranscriptionHandler, handleCustomErrors, videoInferenceDimensionsHandler, videoModelSearchFilterHandler, audioModelSearchFilterHandler, useParameterToggleHandler, imageInferenceToggleHandler, upscalerToggleHandler, videoUpscalerToggleHandler, audioInferenceToggleHandler, acceleratorOptionsToggleHandler, bytedanceProviderSettingsToggleHandler, openaiProviderSettingsToggleHandler, klingProviderSettingsToggleHandler, lumaProviderSettingsToggleHandler, briaProviderSettingsToggleHandler } from "./utils.js";
import { RUNWARE_NODE_TYPES, RUNWARE_NODE_PROPS, SEARCH_TERMS } from "./types.js";
const nodeInitList = [];
@@ -51,6 +51,8 @@ app.registerExtension({
videoInferenceDimensionsHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.UPSCALER) {
upscalerToggleHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.VIDEOUPSCALER) {
videoUpscalerToggleHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.AUDIOINFERENCE) {
audioInferenceToggleHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.ACCELERATOROPTIONS) {
@@ -63,6 +65,8 @@ app.registerExtension({
klingProviderSettingsToggleHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.LUMAPROVIDERSETTINGS) {
lumaProviderSettingsToggleHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.BRIAPROVIDERSETTINGS) {
briaProviderSettingsToggleHandler(node);
}
if(crNodeProps.colorModeOnly === true) return;
clientlibs/types.js
+5
View File
@@ -98,6 +98,7 @@ const RUNWARE_NODE_TYPES = {
KLINGPROVIDERSETTINGS: "Runware KlingAI Provider Settings",
RUNWAYPROVIDERSETTINGS: "Runware Runway Provider Settings",
LUMAPROVIDERSETTINGS: "Runware Luma Provider Settings",
MIDJOURNEYPROVIDERSETTINGS: "Runware Midjourney Provider Settings",
VECTORIZE: "Runware Vectorize",
VIDEOBGREMOVAL: "Runware Video Background Removal",
VIDEOUPSCALER: "Runware Video Upscaler",
@@ -314,6 +315,10 @@ const RUNWARE_NODE_PROPS = {
bgColor: DEFAULT_BGCOLOR,
colorModeOnly: true,
},
[RUNWARE_NODE_TYPES.MIDJOURNEYPROVIDERSETTINGS]: {
bgColor: DEFAULT_BGCOLOR,
colorModeOnly: true,
},
[RUNWARE_NODE_TYPES.VECTORIZE]: {
bgColor: DEFAULT_BGCOLOR,
},
clientlibs/utils.js
+155
View File
@@ -657,6 +657,28 @@ function toggleWidgetEnabled(widget, enabled, node) {
}
}
function videoUpscalerToggleHandler(videoUpscalerNode) {
if (!videoUpscalerNode?.widgets) return;
const useUpscaleFactorWidget = videoUpscalerNode.widgets.find(w => w && w.name === "useUpscaleFactor");
const upscaleFactorWidget = videoUpscalerNode.widgets.find(w => w && w.name === "upscaleFactor");
if (!useUpscaleFactorWidget || !upscaleFactorWidget) return;
function applyToggleState() {
const enabled = useUpscaleFactorWidget.value === true;
toggleWidgetEnabled(upscaleFactorWidget, enabled, videoUpscalerNode);
videoUpscalerNode.setDirtyCanvas(true);
}
// Initialize state once widgets are rendered
setTimeout(applyToggleState, 100);
appendWidgetCB(useUpscaleFactorWidget, () => {
setTimeout(applyToggleState, 50);
});
}
function useParameterToggleHandler(node) {
// Prevent double registration
if (node._useParameterToggleHandlerRegistered) return;
@@ -1931,6 +1953,137 @@ function lumaProviderSettingsToggleHandler(lumaNode) {
}
}
function briaProviderSettingsToggleHandler(briaNode) {
// Find all "use" parameter widgets for Bria Provider Settings
const useMediumWidget = briaNode.widgets.find(w => w.name === "useMedium");
const mediumWidget = briaNode.widgets.find(w => w.name === "medium");
const usePromptEnhancementWidget = briaNode.widgets.find(w => w.name === "usePromptEnhancement");
const promptEnhancementWidget = briaNode.widgets.find(w => w.name === "promptEnhancement");
const useEnhanceImageWidget = briaNode.widgets.find(w => w.name === "useEnhanceImage");
const enhanceImageWidget = briaNode.widgets.find(w => w.name === "enhanceImage");
const usePromptContentModerationWidget = briaNode.widgets.find(w => w.name === "usePromptContentModeration");
const promptContentModerationWidget = briaNode.widgets.find(w => w.name === "promptContentModeration");
const useContentModerationWidget = briaNode.widgets.find(w => w.name === "useContentModeration");
const contentModerationWidget = briaNode.widgets.find(w => w.name === "contentModeration");
const useIpSignalWidget = briaNode.widgets.find(w => w.name === "useIpSignal");
const ipSignalWidget = briaNode.widgets.find(w => w.name === "ipSignal");
const usePreserveAlphaWidget = briaNode.widgets.find(w => w.name === "usePreserveAlpha");
const preserveAlphaWidget = briaNode.widgets.find(w => w.name === "preserveAlpha");
const useModeWidget = briaNode.widgets.find(w => w.name === "useMode");
const modeWidget = briaNode.widgets.find(w => w.name === "mode");
const useEnhanceReferenceImagesWidget = briaNode.widgets.find(w => w.name === "useEnhanceReferenceImages");
const enhanceReferenceImagesWidget = briaNode.widgets.find(w => w.name === "enhanceReferenceImages");
const useRefinePromptWidget = briaNode.widgets.find(w => w.name === "useRefinePrompt");
const refinePromptWidget = briaNode.widgets.find(w => w.name === "refinePrompt");
const useOriginalQualityWidget = briaNode.widgets.find(w => w.name === "useOriginalQuality");
const originalQualityWidget = briaNode.widgets.find(w => w.name === "originalQuality");
const useForceBackgroundDetectionWidget = briaNode.widgets.find(w => w.name === "useForceBackgroundDetection");
const forceBackgroundDetectionWidget = briaNode.widgets.find(w => w.name === "forceBackgroundDetection");
// Helper function to toggle widget enabled state (exact same pattern)
function toggleWidgetState(useWidget, paramWidget, paramName) {
if (!useWidget || !paramWidget) return;
function toggleEnabled() {
const enabled = useWidget.value === true;
// Disable/enable widgets using inputEl if available (exact same pattern)
if (paramWidget.inputEl) {
paramWidget.inputEl.disabled = !enabled;
paramWidget.inputEl.style.opacity = enabled ? "1" : "0.5";
paramWidget.inputEl.style.cursor = enabled ? "text" : "not-allowed";
paramWidget.inputEl.readOnly = !enabled;
}
// Handle dropdown widgets (for medium and mode)
if (paramWidget.options && paramWidget.options.element) {
paramWidget.options.element.disabled = !enabled;
paramWidget.options.element.style.opacity = enabled ? "1" : "0.5";
paramWidget.options.element.style.pointerEvents = enabled ? "auto" : "none";
}
// Also set widget property
paramWidget.disabled = !enabled;
// Fallback: try to find inputs via DOM if inputEl is not available
if (!paramWidget.inputEl) {
const nodeElement = briaNode.htmlElements?.widgetsContainer || briaNode.htmlElements;
if (nodeElement) {
const input = nodeElement.querySelector(`input[name="${paramName}"], textarea[name="${paramName}"], select[name="${paramName}"]`);
if (input) {
input.disabled = !enabled;
input.style.opacity = enabled ? "1" : "0.5";
input.style.cursor = enabled ? "text" : "not-allowed";
input.readOnly = !enabled;
if (input.tagName === "SELECT") {
input.style.pointerEvents = enabled ? "auto" : "none";
}
}
}
}
briaNode.setDirtyCanvas(true);
}
// Set up callback (exact same pattern)
appendWidgetCB(useWidget, () => {
setTimeout(toggleEnabled, 50);
});
// Initial call to set initial state
setTimeout(toggleEnabled, 100);
}
// Set up all toggle handlers
if (useMediumWidget && mediumWidget) {
toggleWidgetState(useMediumWidget, mediumWidget, "medium");
}
if (usePromptEnhancementWidget && promptEnhancementWidget) {
toggleWidgetState(usePromptEnhancementWidget, promptEnhancementWidget, "promptEnhancement");
}
if (useEnhanceImageWidget && enhanceImageWidget) {
toggleWidgetState(useEnhanceImageWidget, enhanceImageWidget, "enhanceImage");
}
if (usePromptContentModerationWidget && promptContentModerationWidget) {
toggleWidgetState(usePromptContentModerationWidget, promptContentModerationWidget, "promptContentModeration");
}
if (useContentModerationWidget && contentModerationWidget) {
toggleWidgetState(useContentModerationWidget, contentModerationWidget, "contentModeration");
}
if (useIpSignalWidget && ipSignalWidget) {
toggleWidgetState(useIpSignalWidget, ipSignalWidget, "ipSignal");
}
if (usePreserveAlphaWidget && preserveAlphaWidget) {
toggleWidgetState(usePreserveAlphaWidget, preserveAlphaWidget, "preserveAlpha");
}
if (useModeWidget && modeWidget) {
toggleWidgetState(useModeWidget, modeWidget, "mode");
}
if (useEnhanceReferenceImagesWidget && enhanceReferenceImagesWidget) {
toggleWidgetState(useEnhanceReferenceImagesWidget, enhanceReferenceImagesWidget, "enhanceReferenceImages");
}
if (useRefinePromptWidget && refinePromptWidget) {
toggleWidgetState(useRefinePromptWidget, refinePromptWidget, "refinePrompt");
}
if (useOriginalQualityWidget && originalQualityWidget) {
toggleWidgetState(useOriginalQualityWidget, originalQualityWidget, "originalQuality");
}
if (useForceBackgroundDetectionWidget && forceBackgroundDetectionWidget) {
toggleWidgetState(useForceBackgroundDetectionWidget, forceBackgroundDetectionWidget, "forceBackgroundDetection");
}
}
export {
notifyUser,
promptEnhanceHandler,
@@ -1947,10 +2100,12 @@ export {
useParameterToggleHandler,
imageInferenceToggleHandler,
upscalerToggleHandler,
videoUpscalerToggleHandler,
audioInferenceToggleHandler,
acceleratorOptionsToggleHandler,
bytedanceProviderSettingsToggleHandler,
openaiProviderSettingsToggleHandler,
klingProviderSettingsToggleHandler,
lumaProviderSettingsToggleHandler,
briaProviderSettingsToggleHandler,
};
modules/providerSettings/briaImageProviderSettings.py
+140 -25
View File
@@ -14,43 +14,119 @@ class RunwareBriaProviderSettings:
return {
"required": {},
"optional": {
"useMedium": ("BOOLEAN", {
"tooltip": "Enable to include medium parameter in API request. Disable if not needed.",
"default": False,
}),
"medium": (["None", "photography", "art"], {
"tooltip": "Select the medium type for image generation",
"tooltip": "Select the medium type for image generation. Only used when 'Use Medium' is enabled.",
"default": "photography",
}),
"usePromptEnhancement": ("BOOLEAN", {
"tooltip": "Enable to include promptEnhancement parameter in API request.",
"default": False,
}),
"promptEnhancement": ("BOOLEAN", {
"tooltip": "When set to true, enhances the provided prompt by generating additional, more descriptive variations, resulting in more diverse and creative output images. Note that turning this flag on may result in a few additional seconds to the inference time.",
"tooltip": "When set to true, enhances the provided prompt by generating additional, more descriptive variations, resulting in more diverse and creative output images. Note that turning this flag on may result in a few additional seconds to the inference time. Only used when 'Use Prompt Enhancement' is enabled.",
"default": False,
"label_on": "Enabled",
"label_off": "Disabled",
}),
"useEnhanceImage": ("BOOLEAN", {
"tooltip": "Enable to include enhanceImage parameter in API request.",
"default": False,
}),
"enhanceImage": ("BOOLEAN", {
"tooltip": "When set to true, generates images with richer details, sharper textures, and enhanced clarity. Slightly increases generation time per image.",
"tooltip": "When set to true, generates images with richer details, sharper textures, and enhanced clarity. Slightly increases generation time per image. Only used when 'Use Enhance Image' is enabled.",
"default": False,
"label_on": "Enabled",
"label_off": "Disabled",
}),
"usePromptContentModeration": ("BOOLEAN", {
"tooltip": "Enable to include promptContentModeration parameter in API request.",
"default": False,
}),
"promptContentModeration": ("BOOLEAN", {
"tooltip": "When enabled (default: true), the input prompt is scanned for NSFW or ethically restricted terms before image generation. If the prompt violates Bria's ethical guidelines, the request will be rejected with a 408 error.",
"tooltip": "When enabled (default: true), the input prompt is scanned for NSFW or ethically restricted terms before image generation. If the prompt violates Bria's ethical guidelines, the request will be rejected with a 408 error. Only used when 'Use Prompt Content Moderation' is enabled.",
"default": True,
"label_on": "Enabled",
"label_off": "Disabled",
}),
"useContentModeration": ("BOOLEAN", {
"tooltip": "Enable to include contentModeration parameter in API request.",
"default": False,
}),
"contentModeration": ("BOOLEAN", {
"tooltip": "When enabled, applies content moderation to both input visuals and generated outputs.",
"tooltip": "When enabled, applies content moderation to both input visuals and generated outputs. Only used when 'Use Content Moderation' is enabled.",
"default": True,
"label_on": "Enabled",
"label_off": "Disabled",
}),
"useIpSignal": ("BOOLEAN", {
"tooltip": "Enable to include ipSignal parameter in API request.",
"default": False,
}),
"ipSignal": ("BOOLEAN", {
"tooltip": "Flags prompts with potential IP content. If detected, a warning will be included in the response.",
"tooltip": "Flags prompts with potential IP content. If detected, a warning will be included in the response. Only used when 'Use IP Signal' is enabled.",
"default": False,
"label_on": "Enabled",
"label_off": "Disabled",
}),
"usePreserveAlpha": ("BOOLEAN", {
"tooltip": "Enable to include preserveAlpha parameter in API request.",
"default": False,
}),
"preserveAlpha": ("BOOLEAN", {
"tooltip": "Preserve alpha channel in the image.",
"default": None,
"tooltip": "Preserve alpha channel in the image. Only used when 'Use Preserve Alpha' is enabled.",
"default": False,
"label_on": "Enabled",
"label_off": "Disabled",
}),
"useMode": ("BOOLEAN", {
"tooltip": "Enable to include mode parameter in API request for Background Replace.",
"default": False,
}),
"mode": (["None", "base", "high_control", "fast"], {
"tooltip": "Selects the background-generation mode for Background Replace. Base: clean, high quality backgrounds. High_control: stronger prompt adherence and scene context, finer control over layout and details. Fast: same core capabilities as base, optimal speed and quality balance. Only used when 'Use Mode' is enabled.",
"default": "base",
}),
"useEnhanceReferenceImages": ("BOOLEAN", {
"tooltip": "Enable to include enhanceReferenceImages parameter in API request.",
"default": False,
}),
"enhanceReferenceImages": ("BOOLEAN", {
"tooltip": "When set to true, additional logic processes the included reference image to make adjustments for optimal results. Only used when 'Use Enhance Reference Images' is enabled.",
"default": True,
"label_on": "Enabled",
"label_off": "Disabled",
}),
"useRefinePrompt": ("BOOLEAN", {
"tooltip": "Enable to include refinePrompt parameter in API request.",
"default": False,
}),
"refinePrompt": ("BOOLEAN", {
"tooltip": "When true, an additional logic takes the prompt that was included and adjusts it to achieve optimal results. Only used when 'Use Refine Prompt' is enabled.",
"default": True,
"label_on": "Enabled",
"label_off": "Disabled",
}),
"useOriginalQuality": ("BOOLEAN", {
"tooltip": "Enable to include originalQuality parameter in API request.",
"default": False,
}),
"originalQuality": ("BOOLEAN", {
"tooltip": "When true, the output image retains the original input image's size; otherwise, the image is scaled to 1 megapixel (1MP) while preserving its aspect ratio. Only used when 'Use Original Quality' is enabled.",
"default": True,
"label_on": "Enabled",
"label_off": "Disabled",
}),
"useForceBackgroundDetection": ("BOOLEAN", {
"tooltip": "Enable to include forceBackgroundDetection parameter in API request.",
"default": False,
}),
"forceBackgroundDetection": ("BOOLEAN", {
"tooltip": "When true, forces background detection and removal, even if the original image already contains an alpha channel. Useful for refining existing foreground/background separation or ignoring unnecessary alpha channels. Only used when 'Use Force Background Detection' is enabled.",
"default": False,
"label_on": "Enabled",
"label_off": "Disabled",
}),
@@ -61,51 +137,90 @@ class RunwareBriaProviderSettings:
RETURN_NAMES = ("providerSettings",)
FUNCTION = "create_provider_settings"
CATEGORY = "Runware/Provider Settings"
DESCRIPTION = "Configure Bria-specific provider settings for image generation including medium type, prompt enhancement, image enhancement, and content moderation options."
DESCRIPTION = "Configure Bria-specific provider settings for image generation and background replacement including medium type, prompt enhancement, image enhancement, content moderation options, and background replace mode settings."
def create_provider_settings(self, **kwargs) -> tuple[Dict[str, Any]]:
"""Create Bria provider settings"""
# Get parameters
# Get use parameters
useMedium = kwargs.get("useMedium", False)
usePromptEnhancement = kwargs.get("usePromptEnhancement", False)
useEnhanceImage = kwargs.get("useEnhanceImage", False)
usePromptContentModeration = kwargs.get("usePromptContentModeration", False)
useContentModeration = kwargs.get("useContentModeration", False)
useIpSignal = kwargs.get("useIpSignal", False)
usePreserveAlpha = kwargs.get("usePreserveAlpha", False)
useMode = kwargs.get("useMode", False)
useEnhanceReferenceImages = kwargs.get("useEnhanceReferenceImages", False)
useRefinePrompt = kwargs.get("useRefinePrompt", False)
useOriginalQuality = kwargs.get("useOriginalQuality", False)
useForceBackgroundDetection = kwargs.get("useForceBackgroundDetection", False)
# Get actual parameters
medium = kwargs.get("medium", "photography")
promptEnhancement = kwargs.get("promptEnhancement", False)
enhanceImage = kwargs.get("enhanceImage", False)
promptContentModeration = kwargs.get("promptContentModeration", True)
contentModeration = kwargs.get("contentModeration", True)
ipSignal = kwargs.get("ipSignal", False)
preserveAlpha = kwargs.get("preserveAlpha", None)
preserveAlpha = kwargs.get("preserveAlpha", False)
mode = kwargs.get("mode", "base")
enhanceReferenceImages = kwargs.get("enhanceReferenceImages", True)
refinePrompt = kwargs.get("refinePrompt", True)
originalQuality = kwargs.get("originalQuality", True)
forceBackgroundDetection = kwargs.get("forceBackgroundDetection", False)
# Build settings dictionary
settings = {}
# Add medium
if medium is not None and medium != "None":
# Add medium only if useMedium is enabled
if useMedium and medium is not None and medium != "None":
settings["medium"] = medium
# Add promptEnhancement
if promptEnhancement is not None:
# Add promptEnhancement only if usePromptEnhancement is enabled
if usePromptEnhancement:
settings["promptEnhancement"] = promptEnhancement
# Add enhanceImage
if enhanceImage is not None:
# Add enhanceImage only if useEnhanceImage is enabled
if useEnhanceImage:
settings["enhanceImage"] = enhanceImage
# Add promptContentModeration
if promptContentModeration is not None:
# Add promptContentModeration only if usePromptContentModeration is enabled
if usePromptContentModeration:
settings["promptContentModeration"] = promptContentModeration
# Add contentModeration
if contentModeration is not None:
# Add contentModeration only if useContentModeration is enabled
if useContentModeration:
settings["contentModeration"] = contentModeration
# Add ipSignal
if ipSignal is not None:
# Add ipSignal only if useIpSignal is enabled
if useIpSignal:
settings["ipSignal"] = ipSignal
# Add preserveAlpha
if preserveAlpha is not None:
# Add preserveAlpha only if usePreserveAlpha is enabled
if usePreserveAlpha:
settings["preserveAlpha"] = preserveAlpha
# Add mode only if useMode is enabled (for Background Replace)
if useMode and mode is not None and mode != "None":
settings["mode"] = mode
# Add enhanceReferenceImages only if useEnhanceReferenceImages is enabled
if useEnhanceReferenceImages:
settings["enhanceReferenceImages"] = enhanceReferenceImages
# Add refinePrompt only if useRefinePrompt is enabled
if useRefinePrompt:
settings["refinePrompt"] = refinePrompt
# Add originalQuality only if useOriginalQuality is enabled
if useOriginalQuality:
settings["originalQuality"] = originalQuality
# Add forceBackgroundDetection only if useForceBackgroundDetection is enabled
if useForceBackgroundDetection:
settings["forceBackgroundDetection"] = forceBackgroundDetection
# Clean up None values
settings = {k: v for k, v in settings.items() if v is not None}
modules/providerSettings/midjourneyProviderSettings.py
+120
View File
@@ -0,0 +1,120 @@
"""
Runware Midjourney Provider Settings Node
Provides Midjourney-specific settings for image generation workflows.
"""
from typing import Any, Dict, Optional, Tuple
class RunwareMidjourneyProviderSettings:
"""Runware Midjourney Provider Settings Node"""
QUALITY_STEP = 0.25
QUALITY_MIN = 0.25
QUALITY_MAX = 2.0
DEFAULT_QUALITY = 1.0
DEFAULT_STYLIZE = 0
STYLIZE_MIN = 0
STYLIZE_MAX = 1000
DEFAULT_CHAOS = 0
CHAOS_MIN = 0
CHAOS_MAX = 100
DEFAULT_WEIRD = 0
WEIRD_MIN = 0
WEIRD_MAX = 3000
NIJI_OPTIONS = [
"", # Allow provider default when omitted
"close",
"0",
"5",
"6",
]
@classmethod
def INPUT_TYPES(cls):
return {
"required": {},
"optional": {
"quality": ("FLOAT", {
"tooltip": "Midjourney quality multiplier (provider key: providerSettings.midjourney.quality).",
"default": cls.DEFAULT_QUALITY,
"min": cls.QUALITY_MIN,
"max": cls.QUALITY_MAX,
"step": cls.QUALITY_STEP,
}),
"stylize": ("INT", {
"tooltip": "Controls how strongly Midjourney's artistic style is applied (provider key: providerSettings.midjourney.stylize).",
"default": cls.DEFAULT_STYLIZE,
"min": cls.STYLIZE_MIN,
"max": cls.STYLIZE_MAX,
}),
"chaos": ("INT", {
"tooltip": "Controls variation and unpredictability (provider key: providerSettings.midjourney.chaos).",
"default": cls.DEFAULT_CHAOS,
"min": cls.CHAOS_MIN,
"max": cls.CHAOS_MAX,
}),
"weird": ("INT", {
"tooltip": "Adds surreal, experimental characteristics (provider key: providerSettings.midjourney.weird).",
"default": cls.DEFAULT_WEIRD,
"min": cls.WEIRD_MIN,
"max": cls.WEIRD_MAX,
}),
"niji": (cls.NIJI_OPTIONS, {
"tooltip": "Select the Midjourney anime-focused rendering engine (provider key: providerSettings.midjourney.niji).",
"default": "close",
}),
}
}
RETURN_TYPES = ("RUNWAREPROVIDERSETTINGS",)
RETURN_NAMES = ("Provider Settings",)
FUNCTION = "create_provider_settings"
CATEGORY = "Runware/Provider Settings"
DESCRIPTION = "Configure Midjourney-specific provider settings such as quality, stylize, chaos, weird, and niji options for image generation."
def create_provider_settings(self, **kwargs) -> Tuple[Optional[Dict[str, Any]],]:
"""Build Midjourney provider settings payload."""
quality = kwargs.get("quality", self.DEFAULT_QUALITY)
stylize = kwargs.get("stylize", self.DEFAULT_STYLIZE)
chaos = kwargs.get("chaos", self.DEFAULT_CHAOS)
weird = kwargs.get("weird", self.DEFAULT_WEIRD)
niji = kwargs.get("niji", "close")
settings: Dict[str, Any] = {}
if quality is not None:
settings["quality"] = float(quality)
if stylize is not None:
settings["stylize"] = int(stylize)
if chaos is not None:
settings["chaos"] = int(chaos)
if weird is not None:
settings["weird"] = int(weird)
if isinstance(niji, str) and niji.strip():
settings["niji"] = niji.strip()
# Clean None values so the inference request only includes populated options.
settings = {k: v for k, v in settings.items() if v is not None}
return (settings if settings else None,)
NODE_CLASS_MAPPINGS = {
"RunwareMidjourneyProviderSettings": RunwareMidjourneyProviderSettings,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"RunwareMidjourneyProviderSettings": "Runware Midjourney Provider Settings",
}
modules/videoModelSearch.py
+4 -7
View File
@@ -252,15 +252,12 @@ class videoModelSearch:
dimensions = self.MODEL_DIMENSIONS.get(modelAirCode, self.DEFAULT_DIMENSIONS)
if useResolution:
width = customWidth if customWidth > 0 else dimensions["width"]
height = customHeight if customHeight > 0 else dimensions["height"]
width = dimensions["width"]
height = dimensions["height"]
if width == 0 and height == 0:
width = None
height = None
else:
width = 0
height = 0
width = None
height = None
return ({
"model": modelAirCode,
modules/videoUpscaler.py
+31 -19
View File
@@ -14,16 +14,22 @@ class videoUpscaler:
"Video": ("STRING", {
"tooltip": "MediaUUID from Runware Media Upload node for the video to upscale."
}),
"upscaleFactor": ([2, 4], {
"tooltip": "Integer scale factor for upscaling (2x or 4x).",
"default": 2,
}),
"Model": (list(cls.RUNWARE_VUPSCALER_MODELS.keys()), {
"tooltip": "Select the video upscaler model to use.",
"default": "Bria Video Upscaler",
}),
"useUpscaleFactor": ("BOOLEAN", {
"label_on": "Yes",
"label_off": "No",
"tooltip": "Enable to explicitly set the upscale factor for the request.",
"default": False,
}),
},
"optional": {
"upscaleFactor": ([2, 4], {
"tooltip": "Integer scale factor for upscaling (2x or 4x).",
"default": 2,
}),
"Output Format": (["mp4", "webm"], {
"default": "mp4",
"tooltip": "Choose the output video format.",
@@ -39,7 +45,8 @@ class videoUpscaler:
def upscale(self, **kwargs):
video_uuid = kwargs.get("Video")
upscaleFactor = kwargs.get("upscaleFactor", 2)
useUpscaleFactor = kwargs.get("useUpscaleFactor", False)
upscaleFactor = kwargs.get("upscaleFactor")
modelName = kwargs.get("Model", "Bria Video Upscaler")
outputFormat = kwargs.get("Output Format", "mp4")
@@ -51,20 +58,25 @@ class videoUpscaler:
modelAIR = self.RUNWARE_VUPSCALER_MODELS.get(modelName, "bria:50@1")
# Build the API request
genConfig = [
{
"taskType": "upscale",
"taskUUID": rwUtils.genRandUUID(),
"upscaleFactor": upscaleFactor,
"inputs": {
"video": video_uuid
},
"outputType": "URL",
"outputFormat": outputFormat.upper(),
"model": modelAIR,
"includeCost": True
}
]
task_payload = {
"taskType": "upscale",
"taskUUID": rwUtils.genRandUUID(),
"inputs": {
"video": video_uuid
},
"outputType": "URL",
"outputFormat": outputFormat.upper(),
"model": modelAIR,
"includeCost": True
}
if useUpscaleFactor:
if upscaleFactor in (2, 4):
task_payload["upscaleFactor"] = upscaleFactor
else:
raise ValueError("upscaleFactor must be 2 or 4 when useUpscaleFactor is enabled.")
genConfig = [task_payload]
try:
# Debug: Print the request being sent
workflows/Runware_Basic_Image_Inference_Midjourney_I2V.json
+1
View File
File diff suppressed because one or more lines are too long
workflows/Runware_Video_Inference_Luma.json
+1 -1
View File
File diff suppressed because one or more lines are too long