logo
Browse Source

update

Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
main
junjie.jiang 2 years ago
parent
commit
37c379fe78
  1. 1
      clip_vision.py

1
clip_vision.py

@ -26,6 +26,7 @@ class Model:
def __init__(self, model_name, device='cpu'):
self.model = clip.create_model(model_name=model_name, pretrained=True, device=device).visual
self.model.eval()
print('Create local model')
def __call__(self, data: 'Tensor'):
return self.model(data)

Loading…
Cancel
Save