From 33ff89fd354d8ec3ab6affb605a79a931b445d99 Mon Sep 17 00:00:00 2001 From: cubiq Date: Sat, 7 Dec 2024 10:40:22 +0100 Subject: [PATCH] interpolate_pos_encoding is not True by default --- segmentation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/segmentation.py b/segmentation.py index c28e44f..55f8701 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, interpolate_pos_encoding=True) + out = model(**inputs) out = out.logits.unsqueeze(1) out = torch.sigmoid(out[0][0]) out = (out > threshold)