Major Release - WAN Studio & Infrastructure Upgrades

Release of WAN Studio.

New node base class for prototyping & consistency.

...

Author: flowstateeng

FS_Constants.py

@@ -15,7 +15,15 @@
 ##
 # CONSTANTS
 ##
+
+# SYSTEM INFO
+SYSTEM_NAME = 'FlowState Creator Suite'
+SYSTEM_VERSION = '0.1.5'
+
+# IMAGE PARAMETERS
 MAX_RESOLUTION=16384
+
+# TYPES
 TOOLTIP_UNDERLINE = 32
 
 

FS_Mappings.py

@@ -20,13 +20,19 @@
 # MAPPINGS
 ##
 NODE_CLASS_MAPPINGS = {
+    'FlowState_SimpleLatent': FlowState_SimpleLatent,
+    'FlowState_VideoCreator': FlowState_VideoCreator,
     'FlowState_LatentSource': FlowState_LatentSource,
     'FlowState_FluxEngine': FlowState_FluxEngine,
+    'FlowState_WANStudio': FlowState_WANStudio,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
+    'FlowState_VideoCreator': '🌊🎥 FlowState Video Creator',
+    'FlowState_SimpleLatent': '🌊👌 FlowState Simple Latent',
     'FlowState_LatentSource': '🌊🌱 FlowState Latent Source',
     'FlowState_FluxEngine': '🌊🚒 FlowState Flux Engine',
+    'FlowState_WANStudio': '🌊🎬 FlowState WAN Studio',
     # 'FlowState_QuickEdit': '🌊 FlowState Quick Edit',
     # 'FlowState_AssetForge': '🌊 FlowState Asset Forge',
 }

FS_Nodes.py

@@ -1,5 +1,6 @@
 # Project: FlowState Creator Suite
-# Description: A collection of custom nodes to solve problems I couldn't find existing nodes for.
+# Description: A node suite designed for professional production, offering a stable, efficient,
+#   and scalable solution that simplifies complex workflows for high-quality, fine-tuned asset creation.
 # Author: Johnathan Chivington
 # Contact: [email protected] | youtube.com/@flowstateeng
 
@@ -14,8 +15,11 @@
 ##
 # NODES
 ##
+from .FlowState_SimpleLatent import *
 from .FlowState_LatentSource import *
 from .FlowState_FluxEngine import *
+from .FlowState_VideoCreator import *
+from .FlowState_WANStudio import *
 
 
 ##

FS_Types.py

@@ -19,6 +19,11 @@
 import nodes
 
 
+
+# -----------------------------------------------------------
+#                    BEGIN GENERIC TYPES
+# -----------------------------------------------------------
+
 ##
 # ANY TYPE
 ##
@@ -72,11 +77,11 @@ def __ne__(self, __value: object) -> bool:
 )})
 
 # IMAGE
-TYPE_IMG_WIDTH = ('INT', {'default': 1024, 'min': 16, 'max': nodes.MAX_RESOLUTION, 'step': 8, 'tooltip': (
+TYPE_IMG_WIDTH = ('INT', {'default': 1024, 'min': 16, 'max': nodes.MAX_RESOLUTION, 'step': 1, 'tooltip': (
     f' Width\n {"-" * TOOLTIP_UNDERLINE}\n'
     f' - Defines the width of the image.\n\n'
 )})
-TYPE_IMG_HEIGHT = ('INT', {'default': 1024, 'min': 16, 'max': nodes.MAX_RESOLUTION, 'step': 8, 'tooltip': (
+TYPE_IMG_HEIGHT = ('INT', {'default': 1024, 'min': 16, 'max': nodes.MAX_RESOLUTION, 'step': 1, 'tooltip': (
     f' Height\n {"-" * TOOLTIP_UNDERLINE}\n'
     f' - Defines the height of the image.\n\n'
 )})
@@ -115,9 +120,13 @@ def __ne__(self, __value: object) -> bool:
         f' - The amount of denoising applied, lower values will maintain the structure of the initial image allowing for image to image sampling.\n\n'
     )
 })
-TYPE_PROMPT_POSITIVE = ('STRING', {'multiline': True, 'default': 'Enter your positive prompt.', 'tooltip': (
+TYPE_PROMPT_POSITIVE = ('STRING', {'multiline': True, 'default': '✅ Describe the image you want the model to create.', 'tooltip': (
+    f' Positive Prompt\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - ✅ Describe the image you want the model to create.\n\n'
+)})
+TYPE_PROMPT_NEGATIVE = ('STRING', {'multiline': True, 'default': '⛔ Describe what you do not want to see in the image.', 'tooltip': (
     f' Positive Prompt\n {"-" * TOOLTIP_UNDERLINE}\n'
-    f' - Positive text prompt describing your desired output.\n\n'
+    f' - ⛔ Describe what you do not want to see in the image.\n\n'
 )})
 
 # MODEL
@@ -186,12 +195,12 @@ def __ne__(self, __value: object) -> bool:
     f' - Used to encode and decode images.\n\n'
 )})
 
-TYPE_CONTROL_NETS_LIST = lambda: (['none'] + CONTROL_NETS_LIST(), {'tooltip': (
+TYPE_CONTROL_NETS_LIST = lambda: (['disabled'] + CONTROL_NETS_LIST(), {'tooltip': (
     f' Control Net List\n {"-" * TOOLTIP_UNDERLINE}\n'
     f' - List of available Control Nets.\n'
     f' - Used to transfer structure of an input image to a generated output image.\n\n'
 )})
-TYPE_LORAS_LIST = lambda: (['none'] + LORAS_LIST(), {'tooltip': (
+TYPE_LORAS_LIST = lambda: (['disabled'] + LORAS_LIST(), {'tooltip': (
     f' LoRA List\n {"-" * TOOLTIP_UNDERLINE}\n'
     f' - List of available Low-Rank Adaptation models.\n'
     f' - Used to transfer a pre-trained style (cyberpunk, anime, photorealism, disney, etc.) to a generated output image.\n\n'
@@ -232,10 +241,67 @@ def __ne__(self, __value: object) -> bool:
 TYPE_IMAGE = ('IMAGE', )
 
 
+
+# -----------------------------------------------------------
+#                    BEGIN FLOWSTATE TYPES
+# -----------------------------------------------------------
+
+
 ##
-# FLOWSTATE CREATOR TYPES
+# FLOWSTATE CREATOR LABELS
 ##
 
+def pad_label(label):
+    width = 100
+    len_label = len(label)
+    num_spaces = width - len_label
+    left_side = num_spaces // 2 - 10
+    right_side = num_spaces - left_side
+    padded_label = ' ' * left_side + '--- ' + label + ' ---' + ' ' * right_side
+    return padded_label
+
+# GENERIC LABELS
+TYPE_FLOWSTATE_LABEL_MODEL = ('STRING', {'default': pad_label('🤖 Model Settings'), 'tooltip': (
+    f' Label\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - This field is not functional. It is just a label for the group of settings below.\n\n'
+)})
+
+TYPE_FLOWSTATE_LABEL_AUG = ('STRING', {'default': pad_label('🔥 Augmentation Settings'), 'tooltip': (
+    f' Label\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - This field is not functional. It is just a label for the group of settings below.\n\n'
+)})
+
+TYPE_FLOWSTATE_LABEL_ENCODER = ('STRING', {'default': pad_label('🔣 Encoder Settings'), 'tooltip': (
+    f' Label\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - This field is not functional. It is just a label for the group of settings below.\n\n'
+)})
+
+TYPE_FLOWSTATE_LABEL_IMAGE = ('STRING', {'default': pad_label('🖼️ Image Settings'), 'tooltip': (
+    f' Label\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - This field is not functional. It is just a label for the group of settings below.\n\n'
+)})
+
+TYPE_FLOWSTATE_LABEL_VIDEO = ('STRING', {'default': pad_label('🎥 Video Settings'), 'tooltip': (
+    f' Label\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - This field is not functional. It is just a label for the group of settings below.\n\n'
+)})
+
+TYPE_FLOWSTATE_LABEL_SAMPLING = ('STRING', {'default': pad_label('🧪 Sampling Settings'), 'tooltip': (
+    f' Label\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - This field is not functional. It is just a label for the group of settings below.\n\n'
+)})
+
+TYPE_FLOWSTATE_LABEL_PROMPT = ('STRING', {'default': pad_label('📝 Prompt(s)'), 'tooltip': (
+    f' Label\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - This field is not functional. It is just a label for the group of settings below.\n\n'
+)})
+
+
+##
+# FLOWSTATE CREATOR GENERIC TYPES
+##
+
+
 # SAGE ATTENTION
 enabled_sage_modes = [
     "disabled",
@@ -261,11 +327,66 @@ def __ne__(self, __value: object) -> bool:
     f' - If using a checkpoint, then the weight_dtype, clip_1_name, clip_2_name & vae_name fields will ignored.\n\n'
 )})
 
-# LATENT SOURCE
+
+# VIDEO
+TYPE_NUM_VIDEO_FRAMES = ('INT', {'default': 48, 'min': 1, 'max': nodes.MAX_RESOLUTION, 'step': 1, 'tooltip': (
+    f' Number of Video Frames\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - The number of frames you want in your final video.\n\n'
+)})
+
+
+##
+# FLOWSTATE VIDEO CREATOR
+##
+TYPE_FPS = ('INT', {'default': 12, 'min': 1, 'max': 120, 'tooltip': (
+    f' Frames Per Second\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - The number of frames per second in the created video.\n\n'
+)})
+TYPE_FRAMES_IN = ('IMAGE', {'tooltip': (
+    f' Video Frames\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - The frames used to create the video.\n\n'
+)})
+TYPE_AUDIO_IN = ('AUDIO', {'tooltip': (
+    f' Video Audio\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - Optional audio to be added to the video.\n\n'
+)})
+TYPE_FILENAME_PREFIX = ('STRING', {'default': 'video/ComyUI', 'tooltip': (
+    f' Filename Prefix\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - The prefix for the file to save.\n'
+    f' - This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes.\n\n'
+)})
+TYPE_VIDEO_FORMAT = (['mp4', 'auto'], {
+    'tooltip': (
+        f' Video Format\n {"-" * TOOLTIP_UNDERLINE}\n'
+        f' - The format to save the video as.\n\n'
+    )
+})
+TYPE_VIDEO_CODEC = (['h264', 'auto'], {
+    'tooltip': (
+        f' Video Codec\n {"-" * TOOLTIP_UNDERLINE}\n'
+        f' - The codec to use for the video.\n\n'
+    )
+})
+
+
+##
+# FLOWSTATE SIMPLE LATENT
+##
+TYPE_SIMPLE_LATENT_INPUT_TYPE = (['Empty Latent', 'Input Image'], {
+    'tooltip': (
+        f' Latent Type\n {"-" * TOOLTIP_UNDERLINE}\n'
+        f' - Your choice of an empty latent (all zeros) or an image as a latent.\n\n'
+    )
+})
+
+
+##
+# FLOWSTATE LATENT SOURCE
+##
 TYPE_LATENT_BATCH_SIZE = ('INT', {'default': 1, 'min': 1, 'max': 4096, 'tooltip': (
-        f' Custom Batch Size\n {"-" * TOOLTIP_UNDERLINE}\n'
-        f' - The number of images you want to generate.\n\n'
-    )})
+    f' Custom Batch Size\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - The number of images you want to generate.\n\n'
+)})
 TYPE_LATENT_SOURCE_INPUT_TYPE = (['Empty Latent', 'Input Image', 'Uploaded Image'], {
     'tooltip': (
         f' Latent Type\n {"-" * TOOLTIP_UNDERLINE}\n'
@@ -296,7 +417,8 @@ def __ne__(self, __value: object) -> bool:
     ], {
     'tooltip': (
         f' Resolution Selector\n {"-" * TOOLTIP_UNDERLINE}\n'
-        f' - Select custom to use the entered width & height, or select a resolution.\n\n'
+        f' - Select "Custom" to use the entered custom_width & custom_height.\n'
+        f' - Select a preset resolution & orientation.\n\n'
     )
 })
 TYPE_LATENT_SOURCE_ORIENTATION = (['Horizontal', 'Vertical'], {
@@ -305,8 +427,81 @@ def __ne__(self, __value: object) -> bool:
         f' - Resolutions given in horizontal orientation. Select vertical to swap resolution aspect ratio.\n\n'
     )
 })
-TYPE_LATENT_SOURCE_OUT = ('LATENT', )
+TYPE_LATENT_SOURCE_OUT = ('LATENT',)
+
+
+##
+# FLOWSTATE FLUX ENGINE
+##
+TYPE_PROMPT_FLUX_ENGINE = ('STRING', {'multiline': True, 'default': '✅ Describe the image you want Flux to create.', 'tooltip': (
+    f' Positive Prompt\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - ✅ Describe the image you want Flux to create.\n\n'
+)})
+TYPE_FLUX_ENGINE_OUT = ('MODEL', 'CLIP', 'VAE', 'IMAGE', 'LATENT')
+
+
+##
+# FLOWSTATE WAN STUDIO
+##
+TYPE_PROMPT_WAN_STUDIO_POSITIVE = ('STRING', {'multiline': True, 'default': '✅ Describe the video you want WAN to create.', 'tooltip': (
+    f' Positive Prompt\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - ✅ Describe the video you want WAN to create.\n\n'
+)})
+TYPE_PROMPT_WAN_STUDIO_NEGATIVE = ('STRING', {'multiline': True, 'default': '⛔ Describe what you do not want to see in the video.', 'tooltip': (
+    f' Positive Prompt\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - ⛔ Describe what you do not want to see in the video.\n\n'
+)})
+TYPE_WAN_STUDIO_STARTING_FRAME = ('IMAGE', {'tooltip': (
+    f' Starting Frame\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - Optionally, select an input image to use as the starting frame.\n\n'
+)})
+TYPE_WAN_STUDIO_RESOLUTION = ([
+    'Custom',
+    'Use Starting Frame Resolution',
+    # HORIZONTAL
+    '1920x1080 - 16:9',
+    '1280x720 - 16:9',
+    '1280x768 - 5:3',
+    '1280x960 - 4:3',
+    '1024x768 - 4:3',
+    '2048x512 - 4:1',
+    '1152x896 - 9:7',
+    '4096x2048 - 2:1',
+    '2048x1024 - 2:1',
+    '1564x670 - 21:9',
+    '2212x948 - 21:9',
+    # SQUARE
+    '4096x4096 - 1:1',
+    '3072x3072 - 1:1',
+    '2048x2048 - 1:1',
+    '1024x1024 - 1:1',
+    '720x720 - 1:1',
+    '512x512 - 1:1'
+    ], {
+    'tooltip': (
+        f' Resolution Selector\n {"-" * TOOLTIP_UNDERLINE}\n'
+        f' - Select "Custom" to use the entered custom_width & custom_height.\n'
+        f' - Select "Use Starting Frame Resolution" to use the resolution of the input image.\n'
+        f' - Select a preset resolution & orientation.\n\n'
+    )
+})
+TYPE_WAN_CLIP_VISION = ('CLIP_VISION', {'tooltip': (
+    f' CLIP Vision Output\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - Optionally, use a CLIP Vision model.\n\n'
+)})
+TYPE_WAN_STUDIO_OUT = ('IMAGE', 'LATENT')
+
 
-# FLUX ENGINE
-TYPE_FLUX_ENGINE_OUT = ('MODEL', 'CLIP', 'VAE', 'IMAGE', 'LATENT', )
+##
+# FLOWSTATE QUICK EDIT
+##
+TYPE_PROMPT_QUICK_EDIT_CHANGES = ('STRING', {'multiline': True, 'default': 'Describe the edits you want Qwen to make.', 'tooltip': (
+    f' Positive Prompt\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - Describe the edits you want Qwen to make.\n\n'
+)})
+TYPE_PROMPT_QUICK_EDIT_REFINE = ('STRING', {'multiline': True, 'default': 'Describe the new image after the edits are made.', 'tooltip': (
+    f' Positive Prompt\n {"-" * TOOLTIP_UNDERLINE}\n'
+    f' - Describe the new image after the edits are made.\n\n'
+)})
+TYPE_QUICK_EDIT_OUT = ('MODEL', 'CLIP', 'VAE', 'IMAGE', 'LATENT')
 

FS_Utils.py

@@ -32,6 +32,9 @@ def get_output_files():
     files = folder_paths.filter_files_content_types(files, ["image"])
     return files
 
+def correct_model_name(name):
+    return name.replace('.sft', '.safetensors')
+
 
 # MODELS
 def get_vae_list():