From e3d1644fef16f1c85bd9f30be4de4a93024b21f2 Mon Sep 17 00:00:00 2001
From: "junjie.jiang" <junjie.jiang@zilliz.com>
Date: Wed, 7 Dec 2022 16:18:04 +0800
Subject: [PATCH] update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
---
 clip_vision.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/clip_vision.py b/clip_vision.py
index 3775cd5..dd341df 100644
--- 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'):