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.