nodes/MinusZoneAI-ComfyUI-Prompt-MZ
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Format toggle for LLamaCPPCLIPTextEncode(LLamaCPP Universal)

Browse Source
This commit is contained in:
Paul Alesius
2025-08-13 17:09:55 +02:00
parent 1dd2cdbb4d
commit a2ef7fdf1a
2 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+2 -1
View File
@@ -235,9 +235,10 @@ class MZ_LLamaCPPCLIPTextEncode:
"required": {
"style_presets": (style_presets, {"default": style_presets[1]}),
"text": ("STRING", {"multiline": True}),
"translate": ([False, True], {"default": False}),
"keep_device": ([False, True], {"default": False}),
"seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}),
"translate": ([False, True], {"default": False}),
"format": ([False, True], {"default": True}),
},
"optional": {
"clip": ("CLIP",),
mz_llama_core_nodes.py
+13 -11
View File
@@ -88,6 +88,7 @@ def llama_cpp_node_encode(args_dict):
keep_device = args_dict.get("keep_device", False)
seed = args_dict.get("seed", -1)
translate_option = args_dict.get("translate", False)
format_option = args_dict.get("format", True)
options["seed"] = seed
options["chat_format"] = chat_format
@@ -201,19 +202,20 @@ def llama_cpp_node_encode(args_dict):
if keep_device is False:
mz_llama_cpp.freed_gpu_memory(model_file=model_file)
# 去除换行
while response.find("\n") != -1:
response = response.replace("\n", " ")
if format_option is True:
# 去除换行
while response.find("\n") != -1:
response = response.replace("\n", " ")
# 句号换成逗号
while response.find(".") != -1:
response = response.replace(".", ",")
# 句号换成逗号
while response.find(".") != -1:
response = response.replace(".", ",")
# 去除多余逗号
while response.find(",,") != -1:
response = response.replace(",,", ",")
while response.find(", ,") != -1:
response = response.replace(", ,", ",")
# 去除多余逗号
while response.find(",,") != -1:
response = response.replace(",,", ",")
while response.find(", ,") != -1:
response = response.replace(", ,", ",")
if translate_option is True:
response = mz_prompt_utils.Utils.prompt_zh_to_en(response)