fix(api): silence controlnet registration warnings

ssube · Dec 22, 2023 · ccff9a3 · ccff9a3
1 parent 2546084
commit ccff9a3
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 15 deletions.
diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py
@@ -44,6 +44,10 @@
     "ignore",
     ".*Converting a tensor to a Python boolean might cause the trace to be incorrect.*",
 )
+warnings.filterwarnings(
+    "ignore",
+    ".*Overwriting tiny_vit_.*",
+)
 
 logger = getLogger(__name__)
 

diff --git a/api/onnx_web/image/source_filter.py b/api/onnx_web/image/source_filter.py
@@ -1,20 +1,29 @@
 # https://github.com/ForserX/StableDiffusionUI/blob/main/data/repo/diffusion_scripts/controlnet_pipe.py
 
-from logging import getLogger
-from os import path
-
-import cv2
-import numpy as np
-import torch
-import transformers
-from controlnet_aux import HEDdetector, MLSDdetector, OpenposeDetector
-from huggingface_hub import snapshot_download
-from PIL import Image, ImageChops, ImageFilter
-
-from ..server.context import ServerContext
-from .ade_palette import ade_palette
-from .laion_face import generate_annotation
-from .noise_source import noise_source_histogram
+import warnings
+
+warnings.filterwarnings(
+    "ignore",
+    ".*Overwriting tiny_vit_.*",
+)
+
+# from https://stackoverflow.com/a/65418681
+if True:  # noqa: E402
+    from logging import getLogger
+    from os import path
+
+    import cv2
+    import numpy as np
+    import torch
+    import transformers
+    from controlnet_aux import HEDdetector, MLSDdetector, OpenposeDetector
+    from huggingface_hub import snapshot_download
+    from PIL import Image, ImageChops, ImageFilter
+
+    from ..server.context import ServerContext
+    from .ade_palette import ade_palette
+    from .laion_face import generate_annotation
+    from .noise_source import noise_source_histogram
 
 logger = getLogger(__name__)