nodes/alpertunga-bile-prompt-generator-comfyui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

text generation is updated

Browse Source
This commit is contained in:
alpertunga-bile
2024-08-11 15:11:34 +03:00
parent c04d90af00
commit 62763d6103
1 changed files with 15 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generator/generate.py
+15 -4
View File
@@ -121,18 +121,29 @@ def get_generated_texts(
gen_texts = []
for result in results:
generated_text = preprocess(prompt + result, preprocess_mode)
generated_text = preprocess(", ".join([prompt, result]), preprocess_mode)
if recursive_level == 0:
gen_texts.append(generated_text)
continue
if is_self_recursive:
for _ in range(0, recursive_level):
result = generator.generate_text(generated_text, gen_args)
generated_text = preprocess(result, preprocess_mode)
generated_text = preprocess(prompt + generated_text, preprocess_mode)
"""
in self recursive, generated_text is not include the prompt string
"""
generated_text = preprocess(
", ".join([prompt, generated_text]), preprocess_mode
)
else:
for _ in range(0, recursive_level):
result = generator.generate_text(generated_text, gen_args)
generated_text += result
generated_text = preprocess(generated_text, preprocess_mode)
generated_text = preprocess(
", ".join([generated_text, result]), preprocess_mode
)
gen_texts.append(generated_text)