Update nodes.py

This commit is contained in:
kijai
2025-11-07 01:09:22 +02:00
parent 3dacd6a719
commit f972b31bf2
+3 -12
View File
@@ -833,20 +833,11 @@ class TextImageEncodeQwenVL():
def add(cls, clip, prompt, image=None): def add(cls, clip, prompt, image=None):
if image is None: if image is None:
images = [] input_images = []
else: else:
samples = image.movedim(-1, 1) input_images = [image[:, :, :, :3]]
total = int(1280 * 720)
scale_by = math.sqrt(total / (samples.shape[3] * samples.shape[2])) tokens = clip.tokenize(prompt, images=input_images)
width = round(samples.shape[3] * scale_by)
height = round(samples.shape[2] * scale_by)
s = common_upscale(samples, width, height, "area", "disabled")
image = s.movedim(1, -1)
images = [image[:, :, :, :3]]
tokens = clip.tokenize(prompt, images=images)
conditioning = clip.encode_from_tokens_scheduled(tokens) conditioning = clip.encode_from_tokens_scheduled(tokens)
print("Qwen-VL embeds shape:", conditioning[0][0].shape) print("Qwen-VL embeds shape:", conditioning[0][0].shape)
return (conditioning[0][0],) return (conditioning[0][0],)