small changes are done

This commit is contained in:
alpertunga-bile
2024-11-17 22:37:06 +03:00
parent b783b28102
commit 302bfd77cc
+13 -9
View File
@@ -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):