Update mask_adapter/sam_maskadapter.py

mask_adapter/sam_maskadapter.py

@@ -188,7 +188,7 @@ class SAMVisualizationDemo(object):
                 select_mask.extend(locs[0].tolist())
         for idx in select_mask:
             select_cls[idx] = class_preds[idx]
-        semseg = torch.einsum("qc,qhw->chw", select_cls.float(), pred_masks.tensor.float())
+        semseg = torch.einsum("qc,qhw->chw", select_cls.float(), pred_masks.tensor.to(text_features).float())
 
         r = semseg
         blank_area = (r[0] == 0)