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):