Remove prefixes bug from video model identifiers

This commit is contained in:
Sirshendu Ganguly
2025-09-01 13:40:10 -04:00
parent 4d74f59820
commit 655118a9a1
4 changed files with 62 additions and 62 deletions
+59 -59
View File
@@ -2,98 +2,98 @@ class videoModelSearch:
# Define the video models by architecture with their default dimensions
VIDEO_MODELS = {
"KlingAI": [
"urn:air:video:klingai:1@2 (KlingAI V1.0 Pro)",
"urn:air:video:klingai:1@1 (KlingAI V1 Standard)",
"urn:air:video:klingai:2@2 (KlingAI V1.5 Pro)",
"urn:air:video:klingai:2@1 (KlingAI V1.5 Standard)",
"urn:air:video:klingai:3@1 (KlingAI V1.6 Standard)",
"urn:air:video:klingai:3@2 (KlingAI V1.6 Pro)",
"urn:air:video:klingai:4@3 (KlingAI V2.1 Master)",
"urn:air:video:klingai:5@1 (KlingAI V2.1 Standard (I2V))",
"urn:air:video:klingai:5@2 (KlingAI V2.1 Pro (I2V))",
"urn:air:video:klingai:5@3 (KlingAI V2.0 Master)",
"klingai:1@2 (KlingAI V1.0 Pro)",
"klingai:1@1 (KlingAI V1 Standard)",
"klingai:2@2 (KlingAI V1.5 Pro)",
"klingai:2@1 (KlingAI V1.5 Standard)",
"klingai:3@1 (KlingAI V1.6 Standard)",
"klingai:3@2 (KlingAI V1.6 Pro)",
"klingai:4@3 (KlingAI V2.1 Master)",
"klingai:5@1 (KlingAI V2.1 Standard (I2V))",
"klingai:5@2 (KlingAI V2.1 Pro (I2V))",
"klingai:5@3 (KlingAI V2.0 Master)",
],
"Veo": [
"urn:air:video:google:2@0 (Veo 2.0)",
"urn:air:video:google:3@0 (Veo 3.0)",
"urn:air:video:google:3@1 (Veo 3.0 Fast)",
"google:2@0 (Veo 2.0)",
"google:3@0 (Veo 3.0)",
"google:3@1 (Veo 3.0 Fast)",
],
"Seedance": [
"urn:air:video:bytedance:2@1 (Seedance 1.0 Pro)",
"urn:air:video:bytedance:1@1 (Seedance 1.0 Lite)",
"bytedance:2@1 (Seedance 1.0 Pro)",
"bytedance:1@1 (Seedance 1.0 Lite)",
],
"MiniMax": [
"urn:air:video:minimax:1@1 (MiniMax 01 Base)",
"urn:air:video:minimax:2@1 (MiniMax 01 Director)",
"urn:air:video:minimax:2@3 (MiniMax I2V 01 Live)",
"urn:air:video:minimax:3@1 (MiniMax 02 Hailuo)",
"minimax:1@1 (MiniMax 01 Base)",
"minimax:2@1 (MiniMax 01 Director)",
"minimax:2@3 (MiniMax I2V 01 Live)",
"minimax:3@1 (MiniMax 02 Hailuo)",
],
"PixVerse": [
"urn:air:video:pixverse:1@1 (PixVerse v3.5)",
"urn:air:video:pixverse:1@2 (PixVerse v4)",
"urn:air:video:pixverse:1@3 (PixVerse v4.5)",
"pixverse:1@1 (PixVerse v3.5)",
"pixverse:1@2 (PixVerse v4)",
"pixverse:1@3 (PixVerse v4.5)",
],
"Vidu": [
"urn:air:video:vidu:1@0 (Vidu Q1 Classic)",
"urn:air:video:vidu:1@1 (Vidu Q1)",
"urn:air:video:vidu:1@5 (Vidu 1.5)",
"urn:air:video:vidu:2@0 (Vidu 2.0)",
"vidu:1@0 (Vidu Q1 Classic)",
"vidu:1@1 (Vidu Q1)",
"vidu:1@5 (Vidu 1.5)",
"vidu:2@0 (Vidu 2.0)",
],
"Wan": [
"urn:air:video:runware:200@1 (Wan 2.1 1.3B)",
"urn:air:video:runware:200@2 (Wan 2.1 14B)",
"runware:200@1 (Wan 2.1 1.3B)",
"runware:200@2 (Wan 2.1 14B)",
],
}
# Model dimensions mapping
MODEL_DIMENSIONS = {
# KlingAI Models
"urn:air:video:klingai:1@2": {"width": 1280, "height": 720}, # KlingAI V1.0 Pro
"urn:air:video:klingai:1@1": {"width": 1280, "height": 720}, # KlingAI V1 Standard
"urn:air:video:klingai:2@2": {"width": 1920, "height": 1080}, # KlingAI V1.5 Pro
"urn:air:video:klingai:2@1": {"width": 1280, "height": 720}, # KlingAI V1.5 Standard
"urn:air:video:klingai:3@1": {"width": 1280, "height": 720}, # KlingAI V1.6 Standard
"urn:air:video:klingai:3@2": {"width": 1920, "height": 1080}, # KlingAI V1.6 Pro
"urn:air:video:klingai:4@3": {"width": 1280, "height": 720}, # KlingAI V2.1 Master
"urn:air:video:klingai:5@1": {"width": 1280, "height": 720}, # KlingAI V2.1 Standard (I2V)
"urn:air:video:klingai:5@2": {"width": 1920, "height": 1080}, # KlingAI V2.1 Pro (I2V)
"urn:air:video:klingai:5@3": {"width": 1920, "height": 1080}, # KlingAI V2.0 Master
"klingai:1@2": {"width": 1280, "height": 720}, # KlingAI V1.0 Pro
"klingai:1@1": {"width": 1280, "height": 720}, # KlingAI V1 Standard
"klingai:2@2": {"width": 1920, "height": 1080}, # KlingAI V1.5 Pro
"klingai:2@1": {"width": 1280, "height": 720}, # KlingAI V1.5 Standard
"klingai:3@1": {"width": 1280, "height": 720}, # KlingAI V1.6 Standard
"klingai:3@2": {"width": 1920, "height": 1080}, # KlingAI V1.6 Pro
"klingai:4@3": {"width": 1280, "height": 720}, # KlingAI V2.1 Master
"klingai:5@1": {"width": 1280, "height": 720}, # KlingAI V2.1 Standard (I2V)
"klingai:5@2": {"width": 1920, "height": 1080}, # KlingAI V2.1 Pro (I2V)
"klingai:5@3": {"width": 1920, "height": 1080}, # KlingAI V2.0 Master
# Veo Models
"urn:air:video:google:2@0": {"width": 1280, "height": 720}, # Veo 2.0
"urn:air:video:google:3@0": {"width": 1280, "height": 720}, # Veo 3.0
"urn:air:video:google:3@1": {"width": 1280, "height": 720}, # Veo 3.0 Fast
"google:2@0": {"width": 1280, "height": 720}, # Veo 2.0
"google:3@0": {"width": 1280, "height": 720}, # Veo 3.0
"google:3@1": {"width": 1280, "height": 720}, # Veo 3.0 Fast
# Seedance Models
"urn:air:video:bytedance:2@1": {"width": 864, "height": 480}, # Seedance 1.0 Pro
"urn:air:video:bytedance:1@1": {"width": 864, "height": 480}, # Seedance 1.0 Lite
"bytedance:2@1": {"width": 864, "height": 480}, # Seedance 1.0 Pro
"bytedance:1@1": {"width": 864, "height": 480}, # Seedance 1.0 Lite
# MiniMax Models
"urn:air:video:minimax:1@1": {"width": 1366, "height": 768}, # MiniMax 01 Base
"urn:air:video:minimax:2@1": {"width": 1366, "height": 768}, # MiniMax 01 Director
"urn:air:video:minimax:2@3": {"width": 1366, "height": 768}, # MiniMax I2V 01 Live
"urn:air:video:minimax:3@1": {"width": 1366, "height": 768}, # MiniMax 02 Hailuo
"minimax:1@1": {"width": 1366, "height": 768}, # MiniMax 01 Base
"minimax:2@1": {"width": 1366, "height": 768}, # MiniMax 01 Director
"minimax:2@3": {"width": 1366, "height": 768}, # MiniMax I2V 01 Live
"minimax:3@1": {"width": 1366, "height": 768}, # MiniMax 02 Hailuo
# PixVerse Models
"urn:air:video:pixverse:1@1": {"width": 640, "height": 360}, # PixVerse v3.5
"urn:air:video:pixverse:1@2": {"width": 640, "height": 360}, # PixVerse v4
"urn:air:video:pixverse:1@3": {"width": 640, "height": 360}, # PixVerse v4.5
"pixverse:1@1": {"width": 640, "height": 360}, # PixVerse v3.5
"pixverse:1@2": {"width": 640, "height": 360}, # PixVerse v4
"pixverse:1@3": {"width": 640, "height": 360}, # PixVerse v4.5
# Vidu Models
"urn:air:video:vidu:1@0": {"width": 1920, "height": 1080}, # Vidu Q1 Classic
"urn:air:video:vidu:1@1": {"width": 1920, "height": 1080}, # Vidu Q1
"urn:air:video:vidu:1@5": {"width": 1920, "height": 1080}, # Vidu 1.5
"urn:air:video:vidu:2@0": {"width": 1920, "height": 1080}, # Vidu 2.0
"vidu:1@0": {"width": 1920, "height": 1080}, # Vidu Q1 Classic
"vidu:1@1": {"width": 1920, "height": 1080}, # Vidu Q1
"vidu:1@5": {"width": 1920, "height": 1080}, # Vidu 1.5
"vidu:2@0": {"width": 1920, "height": 1080}, # Vidu 2.0
# Wan Models
"urn:air:video:runware:200@1": {"width": 853, "height": 480}, # Wan 2.1 1.3B
"urn:air:video:runware:200@2": {"width": 853, "height": 480}, # Wan 2.1 14B
"runware:200@1": {"width": 853, "height": 480}, # Wan 2.1 1.3B
"runware:200@2": {"width": 853, "height": 480}, # Wan 2.1 14B
}
# Models that support seed parameter
SEED_SUPPORTED_MODELS = [
"urn:air:video:runware:200@1", # Wan 2.1 1.3B
"urn:air:video:runware:200@2", # Wan 2.1 14B
"runware:200@1", # Wan 2.1 1.3B
"runware:200@2", # Wan 2.1 14B
]
@classmethod
@@ -123,7 +123,7 @@ class videoModelSearch:
}),
"VideoList": (all_models, {
"tooltip": "Video Model Results Will Show UP Here So You Could Choose From.",
"default": "urn:air:video:klingai:5@3 (KlingAI 2.1)",
"default": "klingai:5@3 (KlingAI V2.0 Master)",
}),
"Use Search Value": ("BOOLEAN", {
"tooltip": "When Enabled, the value you've set in the search input will be used instead.\n\nThis is useful in case the model search API is down or you prefer to set the model manually.",
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
+1 -1
View File
@@ -1 +1 @@
{"id":"00000000-0000-0000-0000-000000000000","revision":0,"last_node_id":22,"last_link_id":22,"nodes":[{"id":6,"type":"SaveVideo","pos":[462.1513671875,86.5411376953125],"size":[270,249.80087280273438],"flags":{},"order":2,"mode":0,"inputs":[{"localized_name":"video","name":"video","type":"VIDEO","link":21},{"localized_name":"filename_prefix","name":"filename_prefix","type":"STRING","widget":{"name":"filename_prefix"},"link":null},{"localized_name":"format","name":"format","type":"COMBO","widget":{"name":"format"},"link":null},{"localized_name":"codec","name":"codec","type":"COMBO","widget":{"name":"codec"},"link":null}],"outputs":[],"properties":{"cnr_id":"comfy-core","ver":"0.3.46","Node name for S&R":"SaveVideo"},"widgets_values":["video/ComfyUI","auto","auto"]},{"id":12,"type":"Runware Video Model Search","pos":[-456.0134582519531,87.07750701904297],"size":[270,170],"flags":{},"order":0,"mode":0,"inputs":[{"localized_name":"Model Search","name":"Model Search","type":"STRING","widget":{"name":"Model Search"},"link":null},{"localized_name":"Model Architecture","name":"Model Architecture","type":"COMBO","widget":{"name":"Model Architecture"},"link":null},{"localized_name":"VideoList","name":"VideoList","type":"COMBO","widget":{"name":"VideoList"},"link":null},{"localized_name":"Use Search Value","name":"Use Search Value","type":"BOOLEAN","widget":{"name":"Use Search Value"},"link":null}],"outputs":[{"localized_name":"Runware Video Model","name":"Runware Video Model","type":"RUNWAREVIDEOMODEL","links":[22]},{"localized_name":"Width","name":"Width","type":"INT","links":null},{"localized_name":"Height","name":"Height","type":"INT","links":null}],"properties":{"cnr_id":"ComfyUI-Runware","ver":"de0a4c2e258a1f3a104d0f9994f9b7ea66e56736","Node name for S&R":"Runware Video Model Search"},"widgets_values":["","Seedance","urn:air:video:bytedance:2@1 (Seedance 1.0 Pro)",false],"bgcolor":"#5345bf"},{"id":19,"type":"Runware Video Inference","pos":[4.869094371795654,81.73494720458984],"size":[400,490],"flags":{},"order":1,"mode":0,"inputs":[{"localized_name":"Model","name":"Model","type":"RUNWAREVIDEOMODEL","link":22},{"localized_name":"firstImage","name":"firstImage","shape":7,"type":"IMAGE","link":null},{"localized_name":"lastImage","name":"lastImage","shape":7,"type":"IMAGE","link":null},{"localized_name":"providerSettings","name":"providerSettings","shape":7,"type":"DICT","link":null},{"localized_name":"positivePrompt","name":"positivePrompt","type":"STRING","widget":{"name":"positivePrompt"},"link":null},{"localized_name":"negativePrompt","name":"negativePrompt","type":"STRING","widget":{"name":"negativePrompt"},"link":null},{"localized_name":"Multi Inference Mode","name":"Multi Inference Mode","type":"BOOLEAN","widget":{"name":"Multi Inference Mode"},"link":null},{"localized_name":"Prompt Weighting","name":"Prompt Weighting","type":"COMBO","widget":{"name":"Prompt Weighting"},"link":null},{"localized_name":"useCustomDimensions","name":"useCustomDimensions","type":"BOOLEAN","widget":{"name":"useCustomDimensions"},"link":null},{"localized_name":"width","name":"width","type":"INT","widget":{"name":"width"},"link":null},{"localized_name":"height","name":"height","type":"INT","widget":{"name":"height"},"link":null},{"localized_name":"duration","name":"duration","type":"INT","widget":{"name":"duration"},"link":null},{"localized_name":"fps","name":"fps","type":"INT","widget":{"name":"fps"},"link":null},{"localized_name":"outputFormat","name":"outputFormat","type":"COMBO","widget":{"name":"outputFormat"},"link":null},{"localized_name":"batchSize","name":"batchSize","type":"INT","widget":{"name":"batchSize"},"link":null},{"localized_name":"includeCost","name":"includeCost","type":"BOOLEAN","widget":{"name":"includeCost"},"link":null},{"localized_name":"seed","name":"seed","type":"INT","widget":{"name":"seed"},"link":null}],"outputs":[{"localized_name":"VIDEO","name":"VIDEO","type":"VIDEO","links":[21]},{"localized_name":"RW-Task","name":"RW-Task","type":"RUNWARETASK","links":null}],"properties":{"cnr_id":"ComfyUI-Runware","ver":"de0a4c2e258a1f3a104d0f9994f9b7ea66e56736","Node name for S&R":"Runware Video Inference"},"widgets_values":["Ant on a leaf.","",false,"Disabled",false,1024,576,5,24,"mp4",1,true,282988745551878,"randomize"],"bgcolor":"#5345bf"}],"links":[[21,19,0,6,0,"VIDEO"],[22,12,0,19,0,"RUNWAREVIDEOMODEL"]],"groups":[],"config":{},"extra":{"ds":{"scale":0.8690867155630664,"offset":[667.374262491862,64.52210987635999]}},"version":0.4}
{"id":"00000000-0000-0000-0000-000000000000","revision":0,"last_node_id":23,"last_link_id":23,"nodes":[{"id":6,"type":"SaveVideo","pos":[462.1513671875,86.5411376953125],"size":[270,249.80087280273438],"flags":{},"order":2,"mode":0,"inputs":[{"localized_name":"video","name":"video","type":"VIDEO","link":21},{"localized_name":"filename_prefix","name":"filename_prefix","type":"STRING","widget":{"name":"filename_prefix"},"link":null},{"localized_name":"format","name":"format","type":"COMBO","widget":{"name":"format"},"link":null},{"localized_name":"codec","name":"codec","type":"COMBO","widget":{"name":"codec"},"link":null}],"outputs":[],"properties":{"cnr_id":"comfy-core","ver":"0.3.46","Node name for S&R":"SaveVideo"},"widgets_values":["video/ComfyUI","auto","auto"]},{"id":19,"type":"Runware Video Inference","pos":[4.869094371795654,81.73494720458984],"size":[400,490],"flags":{},"order":1,"mode":0,"inputs":[{"localized_name":"Model","name":"Model","type":"RUNWAREVIDEOMODEL","link":23},{"localized_name":"firstImage","name":"firstImage","shape":7,"type":"IMAGE","link":null},{"localized_name":"lastImage","name":"lastImage","shape":7,"type":"IMAGE","link":null},{"localized_name":"providerSettings","name":"providerSettings","shape":7,"type":"DICT","link":null},{"localized_name":"positivePrompt","name":"positivePrompt","type":"STRING","widget":{"name":"positivePrompt"},"link":null},{"localized_name":"negativePrompt","name":"negativePrompt","type":"STRING","widget":{"name":"negativePrompt"},"link":null},{"localized_name":"Multi Inference Mode","name":"Multi Inference Mode","type":"BOOLEAN","widget":{"name":"Multi Inference Mode"},"link":null},{"localized_name":"Prompt Weighting","name":"Prompt Weighting","type":"COMBO","widget":{"name":"Prompt Weighting"},"link":null},{"localized_name":"useCustomDimensions","name":"useCustomDimensions","type":"BOOLEAN","widget":{"name":"useCustomDimensions"},"link":null},{"localized_name":"width","name":"width","type":"INT","widget":{"name":"width"},"link":null},{"localized_name":"height","name":"height","type":"INT","widget":{"name":"height"},"link":null},{"localized_name":"duration","name":"duration","type":"INT","widget":{"name":"duration"},"link":null},{"localized_name":"fps","name":"fps","type":"INT","widget":{"name":"fps"},"link":null},{"localized_name":"outputFormat","name":"outputFormat","type":"COMBO","widget":{"name":"outputFormat"},"link":null},{"localized_name":"batchSize","name":"batchSize","type":"INT","widget":{"name":"batchSize"},"link":null},{"localized_name":"includeCost","name":"includeCost","type":"BOOLEAN","widget":{"name":"includeCost"},"link":null},{"localized_name":"seed","name":"seed","type":"INT","widget":{"name":"seed"},"link":null}],"outputs":[{"localized_name":"VIDEO","name":"VIDEO","type":"VIDEO","links":[21]},{"localized_name":"RW-Task","name":"RW-Task","type":"RUNWARETASK","links":null}],"properties":{"cnr_id":"ComfyUI-Runware","ver":"de0a4c2e258a1f3a104d0f9994f9b7ea66e56736","Node name for S&R":"Runware Video Inference"},"widgets_values":["Ant on a leaf.","",false,"Disabled",false,1024,576,5,24,"mp4",1,true,134877469080789,"randomize"],"bgcolor":"#5345bf"},{"id":23,"type":"Runware Video Model Search","pos":[-388.9165344238281,94.34619140625],"size":[270,170],"flags":{},"order":0,"mode":0,"inputs":[{"localized_name":"Model Search","name":"Model Search","type":"STRING","widget":{"name":"Model Search"},"link":null},{"localized_name":"Model Architecture","name":"Model Architecture","type":"COMBO","widget":{"name":"Model Architecture"},"link":null},{"localized_name":"VideoList","name":"VideoList","type":"COMBO","widget":{"name":"VideoList"},"link":null},{"localized_name":"Use Search Value","name":"Use Search Value","type":"BOOLEAN","widget":{"name":"Use Search Value"},"link":null}],"outputs":[{"localized_name":"Runware Video Model","name":"Runware Video Model","type":"RUNWAREVIDEOMODEL","links":[23]},{"localized_name":"Width","name":"Width","type":"INT","links":null},{"localized_name":"Height","name":"Height","type":"INT","links":null}],"properties":{"cnr_id":"ComfyUI-Runware","ver":"4d74f59820ad59af31176536118380f7e9f0374c","Node name for S&R":"Runware Video Model Search"},"widgets_values":["","Seedance","bytedance:2@1 (Seedance 1.0 Pro)",false],"bgcolor":"#5345bf"}],"links":[[21,19,0,6,0,"VIDEO"],[23,23,0,19,0,"RUNWAREVIDEOMODEL"]],"groups":[],"config":{},"extra":{"ds":{"scale":0.8690867155630664,"offset":[668.2237533847492,64.94685532280354]}},"version":0.4}