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)