Merge pull request #81 from Runware/fix-upscaler

Fix upscaleFactor to INT in image upscaler; add Wan 2.6 Flash video model
This commit is contained in:
Sirshendu Ganguly
2026-01-22 22:08:42 -05:00
committed by GitHub
3 changed files with 29 additions and 8 deletions
+8
View File
@@ -1726,6 +1726,7 @@ function videoModelSearchFilterHandler(videoModelSearchNode) {
"runware:200@6 (Wan 2.2)",
"runware:200@8 (Wan 2.2 A14B Animate)",
"alibaba:wan@2.6 (Wan 2.6)",
"alibaba:wan@2.6-flash (Wan 2.6 Flash)",
],
"OpenAI": [
"openai:3@1 (OpenAI Sora 3.1)", "openai:3@0 (OpenAI Sora 3.0)",
@@ -1758,6 +1759,9 @@ function videoModelSearchFilterHandler(videoModelSearchNode) {
"creatify:aurora@fast (Creatify Aurora Avatar Model API (720p))",
"creatify:aurora@0 (Creatify Aurora Avatar Model API (720p))",
],
"Hunyuan": [
"runware:hunyuanvideo@1.5 (HunyuanVideo-1.5)",
],
};
const MODEL_DIMENSIONS = {
@@ -1807,6 +1811,7 @@ function videoModelSearchFilterHandler(videoModelSearchNode) {
"runware:200@6": {"width": 1280, "height": 720},
"runware:200@8": {"width": 1104, "height": 832},
"alibaba:wan@2.6": {"width": 1280, "height": 720},
"alibaba:wan@2.6-flash": {"width": 1280, "height": 720},
"openai:3@1": {"width": 1280, "height": 720},
"openai:3@0": {"width": 1280, "height": 720},
"lightricks:2@0": {"width": 1920, "height": 1080},
@@ -1824,6 +1829,7 @@ function videoModelSearchFilterHandler(videoModelSearchNode) {
"bria:60@1": {"width": 1280, "height": 720},
"creatify:aurora@fast": {"width": 1280, "height": 720},
"creatify:aurora@0": {"width": 1280, "height": 720},
"runware:hunyuanvideo@1.5": {"width": 848, "height": 480},
};
const MODEL_RESOLUTIONS = {
@@ -1873,6 +1879,7 @@ function videoModelSearchFilterHandler(videoModelSearchNode) {
"runware:200@6": "720p",
"runware:200@8": "720p",
"alibaba:wan@2.6": "720p",
"alibaba:wan@2.6-flash": "720p",
"openai:3@1": "720p",
"openai:3@0": "720p",
"lightricks:2@0": "1080p",
@@ -1890,6 +1897,7 @@ function videoModelSearchFilterHandler(videoModelSearchNode) {
"bria:60@1": "720p",
"creatify:aurora@fast": "720p",
"creatify:aurora@0": "720p",
"runware:hunyuanvideo@1.5": "480p",
};
const DEFAULT_DIMENSIONS = {"width": 1024, "height": 576};
+7 -7
View File
@@ -8,12 +8,12 @@ class upscaler:
"Image": ("IMAGE", {
"tooltip": "Specifies the input image to be upscaled."
}),
"upscaleFactor": ("FLOAT", {
"upscaleFactor": ("INT", {
"tooltip": "Each level will increase the size of the image by the corresponding factor.",
"default": 2.0,
"min": 1.0,
"max": 8.0,
"step": 0.1,
"default": 2,
"min": 1,
"max": 8,
"step": 1,
}),
},
"optional": {
@@ -132,7 +132,7 @@ class upscaler:
def upscale(self, **kwargs):
image = kwargs.get("Image")
upscaleFactor = kwargs.get("upscaleFactor", 2.0)
upscaleFactor = int(kwargs.get("upscaleFactor", 2))
# Get toggle parameters
useSteps = kwargs.get("useSteps", False)
@@ -165,7 +165,7 @@ class upscaler:
"taskType": "imageUpscale",
"taskUUID": rwUtils.genRandUUID(),
"inputImage": rwUtils.convertTensor2IMG(image),
"upscaleFactor": upscaleFactor,
"upscaleFactor": int(upscaleFactor),
"outputFormat": rwUtils.OUTPUT_FORMAT,
"outputQuality": rwUtils.OUTPUT_QUALITY,
"outputType": "base64Data",
+14 -1
View File
@@ -61,6 +61,7 @@ class videoModelSearch:
"runware:200@6 (Wan 2.2)",
"runware:200@8 (Wan 2.2 A14B Animate)",
"alibaba:wan@2.6 (Wan 2.6)",
"alibaba:wan@2.6-flash (Wan 2.6 Flash)",
],
"OpenAI": [
"openai:3@1 (OpenAI Sora 3.1)",
@@ -96,6 +97,9 @@ class videoModelSearch:
"creatify:aurora@fast (Creatify Aurora Avatar Model API (720p))",
"creatify:aurora@0 (Creatify Aurora Avatar Model API (720p))",
],
"Hunyuan": [
"runware:hunyuanvideo@1.5 (HunyuanVideo-1.5)",
],
}
# Model dimensions mapping
@@ -158,6 +162,7 @@ class videoModelSearch:
"runware:200@6": {"width": 1280, "height": 720},
"runware:200@8": {"width": 1104, "height": 832},
"alibaba:wan@2.6": {"width": 1280, "height": 720},
"alibaba:wan@2.6-flash": {"width": 1280, "height": 720},
# OpenAI Models
"openai:3@1": {"width": 1280, "height": 720},
@@ -192,6 +197,9 @@ class videoModelSearch:
# Creatify Models
"creatify:aurora@fast": {"width": 1280, "height": 720},
"creatify:aurora@0": {"width": 1280, "height": 720},
# Hunyuan Models
"runware:hunyuanvideo@1.5": {"width": 848, "height": 480},
}
# Model resolutions mapping
@@ -254,6 +262,7 @@ class videoModelSearch:
"runware:200@6": "720p",
"runware:200@8": "720p",
"alibaba:wan@2.6": "720p",
"alibaba:wan@2.6-flash": "720p",
# OpenAI Models
"openai:3@1": "720p",
@@ -288,6 +297,9 @@ class videoModelSearch:
# Creatify Models
"creatify:aurora@fast": "720p",
"creatify:aurora@0": "720p",
# Hunyuan Models
"runware:hunyuanvideo@1.5": "480p",
}
MODEL_ARCHITECTURES = [
@@ -306,7 +318,8 @@ class videoModelSearch:
"Luma",
"Sync",
"Bria",
"Creatify"
"Creatify",
"Hunyuan"
]
DEFAULT_DIMENSIONS = {"width": 1024, "height": 576}