Browse Source
Update
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
1 changed files with
1 additions and
0 deletions
-
vit_image_embedding.py
|
|
@ -39,6 +39,7 @@ class VitImageEmbedding(Operator): |
|
|
super().__init__() |
|
|
super().__init__() |
|
|
sys.path.append(str(Path(__file__).parent)) |
|
|
sys.path.append(str(Path(__file__).parent)) |
|
|
if framework == 'pytorch': |
|
|
if framework == 'pytorch': |
|
|
|
|
|
import pytorch |
|
|
from pytorch.model import Model |
|
|
from pytorch.model import Model |
|
|
self.model = Model(model_name, weights_path) |
|
|
self.model = Model(model_name, weights_path) |
|
|
config = pytorch.resolve_data_config({}, model=self.model._model) |
|
|
config = pytorch.resolve_data_config({}, model=self.model._model) |
|
|
|
