From 48d5a18bd42b6b4e8bae35e26c848e319b63e792 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Tue, 23 Apr 2024 00:52:48 +0300 Subject: [PATCH] Update nodes.py --- nodes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nodes.py b/nodes.py index c4b9ea1..5e0200d 100644 --- a/nodes.py +++ b/nodes.py @@ -2462,7 +2462,7 @@ Segments an image or batch of images using CLIPSeg. for image in images: image = (image* 255).type(torch.uint8) prompt = text - input_prc = processor(text=prompt, images=image, padding="max_length", return_tensors="pt") + input_prc = processor(text=prompt, images=image, return_tensors="pt") # Move the processed input to the device for key in input_prc: input_prc[key] = input_prc[key].to(device)