Browse Source
update
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
1 changed files with
1 additions and
0 deletions
-
clip_vision.py
|
|
@ -26,6 +26,7 @@ class Model: |
|
|
def __init__(self, model_name, device='cpu'): |
|
|
def __init__(self, model_name, device='cpu'): |
|
|
self.model = clip.create_model(model_name=model_name, pretrained=True, device=device).visual |
|
|
self.model = clip.create_model(model_name=model_name, pretrained=True, device=device).visual |
|
|
self.model.eval() |
|
|
self.model.eval() |
|
|
|
|
|
print('Create local model') |
|
|
|
|
|
|
|
|
def __call__(self, data: 'Tensor'): |
|
|
def __call__(self, data: 'Tensor'): |
|
|
return self.model(data) |
|
|
return self.model(data) |
|
|
|
