small changes are done
This commit is contained in:
+13
-9
@@ -24,7 +24,7 @@ class PromptGenerator:
|
|||||||
_gen_settings = GenerateArgs # gen configurations from the last generation
|
_gen_settings = GenerateArgs # gen configurations from the last generation
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def INPUT_TYPES(s):
|
def INPUT_TYPES(self):
|
||||||
quantize_sizes = get_usable_quantize_sizes()
|
quantize_sizes = get_usable_quantize_sizes()
|
||||||
model_names = [
|
model_names = [
|
||||||
file
|
file
|
||||||
@@ -57,7 +57,7 @@ class PromptGenerator:
|
|||||||
{
|
{
|
||||||
"default": 1.0,
|
"default": 1.0,
|
||||||
"min": 0.0,
|
"min": 0.0,
|
||||||
"max": INT_MAX,
|
"max": FLOAT_MAX,
|
||||||
"step": 0.1,
|
"step": 0.1,
|
||||||
},
|
},
|
||||||
),
|
),
|
||||||
@@ -104,7 +104,7 @@ class PromptGenerator:
|
|||||||
"self_recursive": (["disable", "enable"],),
|
"self_recursive": (["disable", "enable"],),
|
||||||
"recursive_level": (
|
"recursive_level": (
|
||||||
"INT",
|
"INT",
|
||||||
{"default": 0, "min": 0, "max": FLOAT_MAX, "step": 1},
|
{"default": 0, "min": 0, "max": INT_MAX, "step": 1},
|
||||||
),
|
),
|
||||||
"preprocess_mode": (["exact_keyword", "exact_prompt", "none"],),
|
"preprocess_mode": (["exact_keyword", "exact_prompt", "none"],),
|
||||||
},
|
},
|
||||||
@@ -138,7 +138,7 @@ class PromptGenerator:
|
|||||||
file.write(f"Date & Time : {datetime.now()}\n")
|
file.write(f"Date & Time : {datetime.now()}\n")
|
||||||
file.write(f"Model : {model_name}\n")
|
file.write(f"Model : {model_name}\n")
|
||||||
file.write(f"Prompt : {prompt}\n")
|
file.write(f"Prompt : {prompt}\n")
|
||||||
file.write(f"Generated Prompts :\n")
|
file.write("Generated Prompts :\n")
|
||||||
|
|
||||||
for i in range(len(self._generated_prompts)):
|
for i in range(len(self._generated_prompts)):
|
||||||
file.write(
|
file.write(
|
||||||
@@ -236,10 +236,12 @@ class PromptGenerator:
|
|||||||
|
|
||||||
is_lock_generation = True if lock == "enable" else False
|
is_lock_generation = True if lock == "enable" else False
|
||||||
|
|
||||||
# check if it is the first generation with taking length of tokenized prompts
|
"""
|
||||||
# and the boolean with is lock enabled
|
check if it is the first generation with taking length of tokenized prompts
|
||||||
# if it is true just return from the lists with assigned new index (declaration is above)
|
and the boolean with is lock enabled
|
||||||
# log the outputs for the clearity
|
if it is true just return from the lists with assigned new index (declaration is above)
|
||||||
|
log the outputs for the clearity
|
||||||
|
"""
|
||||||
if is_lock_generation is True and len(self._tokenized_prompts) > 0:
|
if is_lock_generation is True and len(self._tokenized_prompts) > 0:
|
||||||
self.__log_outputs(
|
self.__log_outputs(
|
||||||
model_name,
|
model_name,
|
||||||
@@ -325,7 +327,9 @@ class PromptGenerator:
|
|||||||
)
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def VALIDATE_INPUTS(s, clip: CLIP, model_name: str):
|
def VALIDATE_INPUTS(self, **kwargs):
|
||||||
|
model_name = kwargs["model_name"]
|
||||||
|
|
||||||
model_path = join(models_dir, "prompt_generators", model_name)
|
model_path = join(models_dir, "prompt_generators", model_name)
|
||||||
|
|
||||||
if not exists(model_path):
|
if not exists(model_path):
|
||||||
|
|||||||
Reference in New Issue
Block a user