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