fix for clipseg, note needs up-to-date transformers

2024-11-20 15:31:13 +01:00
parent 64e38fd0f3
commit 5a4e5cd575
1 changed files with 1 additions and 1 deletions
@@ -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)