From bd91cadaa5e116e959c112444aaaf6f558c7376a Mon Sep 17 00:00:00 2001 From: SeanScripts Date: Wed, 25 Sep 2024 16:04:55 -0700 Subject: [PATCH] Fix dumb typo --- nodes.py | 4 ++-- pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index c2d75dc..3451c93 100644 --- a/nodes.py +++ b/nodes.py @@ -142,7 +142,7 @@ class LlamaVisionGenerateText: "required": { "llama_vision_model": ("LLAMA_VISION_MODEL",), "images": ("IMAGE",), - "prompt": ("STRING", {"default": "<|begin_of_text|><|start_header_id|>user<|end_header_id|>\nCaption this image:\n<|image|><|eot_id|><|start_header_id|>assistant<|end_header_id|>", "multiline": True}), + "prompt": ("STRING", {"default": "<|begin_of_text|><|start_header_id|>user<|end_header_id|>\n\nCaption this image:\n<|image|><|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n", "multiline": True}), "max_new_tokens": ("INT", {"default": 256, "min": 1, "max": 4096}), "do_sample": ("BOOLEAN", {"default": True}), "temperature": ("FLOAT", {"default": 0.5}), @@ -179,7 +179,7 @@ class LlamaVisionGenerateText: time_per_token = generated_tokens/total_time print(f"Generated {generated_tokens} tokens in {total_time:.3f} s ({time_per_token:.3f} tok/s)") print(len(generate_ids[0][prompt_tokens:])) - output = pixtral_model['processor'].decode(generate_ids[0][prompt_tokens:], skip_special_tokens=True, clean_up_tokenization_spaces=False) + output = llama_vision_model['processor'].decode(generate_ids[0][prompt_tokens:], skip_special_tokens=True, clean_up_tokenization_spaces=False) print(output) return (output,) diff --git a/pyproject.toml b/pyproject.toml index 8f83a1f..684e120 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "comfyui-pixtralllamavision" description = "For loading and running Pixtral and Llama 3.2 Vision models" -version = "2.0.0" +version = "2.0.1" license = {file = "LICENSE"} [project.urls]