1580158
/
sd-webui-inpaint-anything
mirror of https://github.com/Uminosachi/sd-webui-inpaint-anything.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove UserWarning when registering TinyViT model

main
Uminosachi 2024-06-21 09:14:06 +09:00
parent df7bfd8b2c
commit 798737ae78
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ia_logging.py
View File

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

10
mobile_sam/modeling/tiny_vit_sam.py
View File

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