update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
3 years ago · e3d1644fef
1 changed files with 4 additions and 1 deletions
--- a/clip_vision.py
+++ b/clip_vision.py
@ -49,7 +49,10 @@ class ClipVision(NNOperator):
    @property
    def device(self):
        if self._device is None:
-            self._device = torch.device(self._device_id)
+            if self._device_id < 0:
+                self._device = torch.device('cpu')
+            else:
+                self._device = torch.device(self._device_id)
        return self._device

    def __call__(self, image: 'Image'):