From ccff9a3136635c580e9471474c0ad17af6a59039 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 21 Dec 2023 21:30:49 -0600 Subject: [PATCH] fix(api): silence controlnet registration warnings --- api/onnx_web/convert/__main__.py | 4 ++++ api/onnx_web/image/source_filter.py | 35 ++++++++++++++++++----------- 2 files changed, 26 insertions(+), 13 deletions(-) diff --git a/api/onnx_web/convert/__main__.py b/api/onnx_web/convert/__main__.py index 0b68ecc1..27d0c42d 100644 --- a/api/onnx_web/convert/__main__.py +++ b/api/onnx_web/convert/__main__.py @@ -44,6 +44,10 @@ warnings.filterwarnings( "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 index 9ad9237f..d024d39e 100644 --- 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 warnings -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 +warnings.filterwarnings( + "ignore", + ".*Overwriting tiny_vit_.*", +) -from ..server.context import ServerContext -from .ade_palette import ade_palette -from .laion_face import generate_annotation -from .noise_source import noise_source_histogram +# 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__)