1
0
Fork 0

fix(api): silence controlnet registration warnings

This commit is contained in:
Sean Sube 2023-12-21 21:30:49 -06:00
parent 2546084208
commit ccff9a3136
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 26 additions and 13 deletions

View File

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

View File

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