From 37c379fe78d33a527be35f5bd68979a5b3a47525 Mon Sep 17 00:00:00 2001 From: "junjie.jiang" Date: Wed, 7 Dec 2022 15:58:02 +0800 Subject: [PATCH] update Signed-off-by: junjie.jiang --- clip_vision.py | 1 + 1 file changed, 1 insertion(+) diff --git a/clip_vision.py b/clip_vision.py index 273593e..3775cd5 100644 --- a/clip_vision.py +++ b/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)