From bf39483d37a000ff5319dea2fc1db7ae39e8a1f9 Mon Sep 17 00:00:00 2001 From: wailovet Date: Tue, 23 Apr 2024 03:04:59 +0800 Subject: [PATCH] 'update' --- __init__.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/__init__.py b/__init__.py index ef20721..6b67af8 100644 --- a/__init__.py +++ b/__init__.py @@ -185,7 +185,7 @@ class MZ_LLavaImageInterrogator: ), "image": ("IMAGE",), "resolution": ("INT", {"default": 512, "min": 128, "max": 2048}), - "prefix" : ("STRING", {"default": "(masterpiece)", "multiline": True, }), + # "prefix" : ("STRING", {"default": "(masterpiece)", "multiline": True, }), "seed": ("INT", {"default": 0, "min": 0, "max": 0xffffffffffffffff}), "n_gpu_layers": ("INT", {"default": 40, "min": -1, "max": 0xffffffffffffffff}), }, @@ -196,15 +196,15 @@ class MZ_LLavaImageInterrogator: RETURN_TYPES = ("STRING", "CONDITIONING",) FUNCTION = "interrogate" CATEGORY = CATEGORY_NAME - def interrogate(self, llama_cpp_model, mmproj_model, image, resolution, prefix, download_source=None, seed=0, clip=None, n_gpu_layers=40, llama_cpp_options=None): + def interrogate(self, llama_cpp_model, mmproj_model, image, resolution, download_source=None, seed=0, clip=None, n_gpu_layers=40, llama_cpp_options=None): importlib.reload(mz_llava) llama_cpp_model = llama_cpp_model.replace("[downloaded]", "") mmproj_model = mmproj_model.replace("[downloaded]", "") - prefix = Utils.prompt_zh_to_en(prefix) - if not prefix.endswith(","): - prefix += "," + # prefix = Utils.prompt_zh_to_en(prefix) + # if not prefix.endswith(","): + # prefix += "," image_pil = Utils.tensor2pil(image)