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)