From 43e89b67b636e0e43793e1848187cf52e3938dc9 Mon Sep 17 00:00:00 2001
From: shiyu22 <shiyu.chen@zilliz.com>
Date: Thu, 16 Dec 2021 12:25:31 +0800
Subject: [PATCH] Update

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
---
 vit_image_embedding.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/vit_image_embedding.py b/vit_image_embedding.py
index 9df2437..8812503 100644
--- a/vit_image_embedding.py
+++ b/vit_image_embedding.py
@@ -39,6 +39,7 @@ class VitImageEmbedding(Operator):
         super().__init__()
         sys.path.append(str(Path(__file__).parent))
         if framework == 'pytorch':
+            import pytorch
             from pytorch.model import Model
         self.model = Model(model_name, weights_path)
         config = pytorch.resolve_data_config({}, model=self.model._model)