From 5a4e5cd5750515d76f3a3363b7d2a0f362507796 Mon Sep 17 00:00:00 2001 From: cubiq Date: Wed, 20 Nov 2024 15:31:13 +0100 Subject: [PATCH] fix for clipseg, note needs up-to-date transformers --- segmentation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/segmentation.py b/segmentation.py index 55f8701..c28e44f 100644 --- a/segmentation.py +++ b/segmentation.py @@ -48,7 +48,7 @@ class ApplyCLIPSeg: outputs = [] for i in imagenp: inputs = processor(text=prompt, images=[i], return_tensors="pt") - out = model(**inputs) + out = model(**inputs, interpolate_pos_encoding=True) out = out.logits.unsqueeze(1) out = torch.sigmoid(out[0][0]) out = (out > threshold)