Browse Source
Fix device
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
1 changed files with
1 additions and
1 deletions
-
clip.py
|
|
@ -107,7 +107,7 @@ class Clip(NNOperator): |
|
|
if model_name in config: |
|
|
if model_name in config: |
|
|
real_name = config[model_name] |
|
|
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.tokenizer = CLIPTokenizer.from_pretrained(real_name) |
|
|
self.processor = CLIPProcessor.from_pretrained(real_name) |
|
|
self.processor = CLIPProcessor.from_pretrained(real_name) |
|
|
|
|
|
|
|
|
|
