Fix device

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
8 months ago · 9fc86c820b
1 changed files with 1 additions and 1 deletions
--- a/clip.py
+++ b/clip.py
@ -107,7 +107,7 @@ class Clip(NNOperator):
        if model_name in config:
            real_name = config[model_name]

-        self.model = Model(real_name, modality, checkpoint_path, device)
+        self.model = Model(real_name, modality, checkpoint_path, self.device)
        self.tokenizer = CLIPTokenizer.from_pretrained(real_name)
        self.processor =  CLIPProcessor.from_pretrained(real_name)