From 9fc86c820bf581f825e98e179261993f795d5dcb Mon Sep 17 00:00:00 2001 From: "junjie.jiang" Date: Wed, 17 Apr 2024 16:06:11 +0800 Subject: [PATCH] Fix device Signed-off-by: junjie.jiang --- clip.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clip.py b/clip.py index cf110dc..a611d5f 100644 --- 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)