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:
@@ -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
@@ -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",
|
||||
|
||||
@@ -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}
|
||||
|
||||
Reference in New Issue
Block a user