added scale factor 1,2

This commit is contained in:
OuticNZ
2024-06-03 08:12:33 +12:00
parent 4b15f0975b
commit 64e71a12ec
+12 -6
View File
@@ -64,6 +64,8 @@ class SoCParameters2Pipe:
"CFG": FLOAT_INPUT,
"image_width": INT_INPUT,
"image_height": INT_INPUT,
"stage1_scale_factor": FLOAT_INPUT,
"stage2_scale_factor": FLOAT_INPUT
}
}
@@ -72,7 +74,7 @@ class SoCParameters2Pipe:
FUNCTION = "execute"
def execute(self, SoCPipeParameters=None, model_name=None, path_name=None, sampler=None, scheduler=None,scheduler_name=None, postive_prompt=None, negative_prompt=None, seed=None, batch_count=None, steps=None, CFG=None, image_width=None, image_height=None):
def execute(self, SoCPipeParameters=None, model_name=None, path_name=None, sampler=None, scheduler=None,scheduler_name=None, postive_prompt=None, negative_prompt=None, seed=None, batch_count=None, steps=None, CFG=None, image_width=None, image_height=None, stage1_scale_factor=None, stage2_scale_factor=None):
model_name_original = None
path_name_original = None
sampler_original = None
@@ -86,9 +88,11 @@ class SoCParameters2Pipe:
CFG_original = None
image_width_original = None
image_height_original = None
stage1_scale_factor_original = None
stage2_scale_factor_original = None
if SoCPipeParameters != None:
model_name_original, path_name_original, sampler_original, scheduler_original, scheduler_name_original, postive_prompt_original, negative_prompt_original,seed_original,batch_count_original,steps_original,CFG_original,image_width_original, image_height_original = SoCPipeParameters
model_name_original, path_name_original, sampler_original, scheduler_original, scheduler_name_original, postive_prompt_original, negative_prompt_original,seed_original,batch_count_original,steps_original,CFG_original,image_width_original, image_height_original, stage1_scale_factor_original, stage2_scale_factor_original = SoCPipeParameters
SoCPipeParametersMod = []
@@ -105,6 +109,8 @@ class SoCParameters2Pipe:
SoCPipeParametersMod.append(CFG if CFG is not None else CFG_original)
SoCPipeParametersMod.append(image_width if image_width is not None else image_width_original)
SoCPipeParametersMod.append(image_height if image_height is not None else image_height_original)
SoCPipeParametersMod.append(stage1_scale_factor if stage1_scale_factor is not None else stage1_scale_factor_original)
SoCPipeParametersMod.append(stage2_scale_factor if stage2_scale_factor is not None else stage2_scale_factor_original)
return (SoCPipeParametersMod,)
@@ -124,11 +130,11 @@ class SoCPipe2Parameters:
}
CATEGORY = "SimpleOfComplex/Pipe"
RETURN_TYPES = (CLASSES.SOC_PIPE_PARAMETERS.value, STRING, STRING, any, any, any, STRING, STRING, INT, INT, INT, FLOAT, INT, INT,)
RETURN_NAMES = ("SoCPipeParameters", "modelname", "path_name", "sampler_name", "scheduler", "scheduler_name", "postive_prompt", "negative_prompt", "seed", "batch_count", "steps", "CFG", "image_height", "image_width")
RETURN_TYPES = (CLASSES.SOC_PIPE_PARAMETERS.value, STRING, STRING, any, any, any, STRING, STRING, INT, INT, INT, FLOAT, INT, INT,FLOAT, FLOAT)
RETURN_NAMES = ("SoCPipeParameters", "modelname", "path_name", "sampler_name", "scheduler", "scheduler_name", "postive_prompt", "negative_prompt", "seed", "batch_count", "steps", "CFG", "image_height", "image_width", "stage1_scale_factor", "stage2_scale_factor")
FUNCTION = "execute"
def execute(self, SoCPipeParameters=None, ):
model_name_original, path_name_original, sampler_original, scheduler_original, postive_prompt_original, negative_prompt_original,seed_original,batch_count_original,steps_original,CFG_original,image_width_original, image_height_original = SoCPipeParameters
return SoCPipeParameters, model_name_original, path_name_original, sampler_original, scheduler_original, postive_prompt_original, negative_prompt_original,seed_original,batch_count_original,steps_original,CFG_original,image_width_original, image_height_original
model_name_original, path_name_original, sampler_original, scheduler_original, postive_prompt_original, negative_prompt_original,seed_original,batch_count_original,steps_original,CFG_original,image_width_original, image_height_original, stage1_scale_factor_original, stage2_scale_factor_original = SoCPipeParameters
return SoCPipeParameters, model_name_original, path_name_original, sampler_original, scheduler_original, postive_prompt_original, negative_prompt_original,seed_original,batch_count_original,steps_original,CFG_original,image_width_original, image_height_original, stage1_scale_factor_original, stage2_scale_factor_original