Spaces:

microsoft
/

visual_chatgpt

Runtime error

LanHarmony commited on Mar 11, 2023

Commit

40bec0d

•

1 Parent(s): 9c259e8

half precision

Files changed (1) hide show

visual_foundation_models.py CHANGED Viewed

@@ -107,7 +107,7 @@ def get_new_image_name(org_img_name, func_name="update"):
 class MaskFormer:
     def __init__(self, device):
         self.device = device
-        self.processor = CLIPSegProcessor.from_pretrained("CIDAS/clipseg-rd64-refined")
         self.model = CLIPSegForImageSegmentation.from_pretrained("CIDAS/clipseg-rd64-refined", torch_dtype=torch.float16).to(device)
     def inference(self, image_path, text):

 class MaskFormer:
     def __init__(self, device):
         self.device = device
+        self.processor = CLIPSegProcessor.from_pretrained("CIDAS/clipseg-rd64-refined", torch_dtype=torch.float16)
         self.model = CLIPSegForImageSegmentation.from_pretrained("CIDAS/clipseg-rd64-refined", torch_dtype=torch.float16).to(device)
     def inference(self, image_path, text):