interpolate_pos_encoding is not True by default

This commit is contained in:
cubiq
2024-12-07 10:40:22 +01:00
parent 5a4e5cd575
commit 33ff89fd35
+1 -1
View File
@@ -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)