image-text-embedding
copied
update
Signed-off-by: junjie.jiang <junjie.jiang@zilliz.com>
@ -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)