diff --git a/__init__.py b/__init__.py index c92edd7..f36ce8a 100644 --- a/__init__.py +++ b/__init__.py @@ -248,8 +248,8 @@ class MZ_LLamaCPPCLIPTextEncode: }, } - RETURN_TYPES = ("STRING", "CONDITIONING",) - RETURN_NAMES = ("text", "conditioning",) + RETURN_TYPES = ("STRING", "CONDITIONING", "STRING") + RETURN_NAMES = ("text", "conditioning","text_raw") OUTPUT_NODE = True FUNCTION = "encode" CATEGORY = CATEGORY_NAME diff --git a/mz_llama_core_nodes.py b/mz_llama_core_nodes.py index f45399a..caaaf3d 100644 --- a/mz_llama_core_nodes.py +++ b/mz_llama_core_nodes.py @@ -235,7 +235,7 @@ def llama_cpp_node_encode(args_dict): conditionings = mz_prompt_utils.Utils.a1111_clip_text_encode( clip, response, ) - return {"ui": {"string": [mz_prompt_utils.Utils.to_debug_prompt(response, translate_option),]}, "result": (response, conditionings)} + return {"ui": {"string": [mz_prompt_utils.Utils.to_debug_prompt(response, translate_option),]}, "result": (response, conditionings, response)} def image_interrogator_captioner(args_dict):