From 558a2375b9a079774d028fcfec24bdf0c37012c4 Mon Sep 17 00:00:00 2001 From: zhang chen Date: Mon, 27 Dec 2021 20:45:20 +0800 Subject: [PATCH] cancel import warning --- vit_image_embedding.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/vit_image_embedding.py b/vit_image_embedding.py index c48566e..310d435 100644 --- a/vit_image_embedding.py +++ b/vit_image_embedding.py @@ -25,6 +25,9 @@ from towhee.operator import Operator from timm.data import resolve_data_config from timm.data.transforms_factory import create_transform +import warnings +warnings.filterwarnings("ignore") + class VitImageEmbedding(Operator): """ Embedding extractor using ViT.