Remove UserWarning when registering TinyViT model

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)

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