Update nodes.py

2025-11-08 12:21:11 +02:00
parent ff16dce5c0
commit 1e00c8fb28
1 changed files with 4 additions and 1 deletions
@@ -834,10 +834,13 @@ class TextImageEncodeQwenVL():
    def add(cls, clip, prompt, image=None):
        if image is None:
            input_images = []
+            llama_template = None
        else:
            input_images = [image[:, :, :, :3]]

-        tokens = clip.tokenize(prompt, images=input_images)
+            llama_template = "<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n<|im_start|>user\n<|vision_start|><|image_pad|><|vision_end|>{}<|im_end|>\n<|im_start|>assistant\n"
+
+        tokens = clip.tokenize(prompt, images=input_images, llama_template=llama_template)
        conditioning = clip.encode_from_tokens_scheduled(tokens)
        print("Qwen-VL embeds shape:", conditioning[0][0].shape)
        return (conditioning[0][0],)