From 798737ae7850445fdc3ea0b25d2cd4059e866fa0 Mon Sep 17 00:00:00 2001
From: Uminosachi <49424133+Uminosachi@users.noreply.github.com>
Date: Fri, 21 Jun 2024 09:14:06 +0900
Subject: [PATCH] Remove UserWarning when registering TinyViT model

---
 ia_logging.py                       |  1 -
 mobile_sam/modeling/tiny_vit_sam.py | 10 +++++++---
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/ia_logging.py b/ia_logging.py
index 477bcbf..c8c3fa2 100644
--- a/ia_logging.py
+++ b/ia_logging.py
@@ -7,7 +7,6 @@ from PIL import Image, ImageDraw
 
 warnings.filterwarnings(action="ignore", category=FutureWarning, module="transformers")
 warnings.filterwarnings(action="ignore", category=FutureWarning, module="huggingface_hub")
-warnings.filterwarnings(action="ignore", category=FutureWarning, module="timm")
 
 ia_logging = logging.getLogger("Inpaint Anything")
 ia_logging.setLevel(logging.INFO)
diff --git a/mobile_sam/modeling/tiny_vit_sam.py b/mobile_sam/modeling/tiny_vit_sam.py
index 4da8988..79205db 100644
--- a/mobile_sam/modeling/tiny_vit_sam.py
+++ b/mobile_sam/modeling/tiny_vit_sam.py
@@ -8,14 +8,18 @@
 # --------------------------------------------------------
 
 import itertools
+import warnings
+from typing import Tuple
+
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
 import torch.utils.checkpoint as checkpoint
-from timm.models.layers import DropPath as TimmDropPath,\
-    to_2tuple, trunc_normal_
+from timm.models.layers import DropPath as TimmDropPath
+from timm.models.layers import to_2tuple, trunc_normal_
 from timm.models.registry import register_model
-from typing import Tuple
+
+warnings.simplefilter("ignore", category=UserWarning)
 
 
 class Conv2d_BN(torch.nn.Sequential):