fix(api): silence controlnet registration warnings
This commit is contained in:
parent
2546084208
commit
ccff9a3136
|
@ -44,6 +44,10 @@ warnings.filterwarnings(
|
||||||
"ignore",
|
"ignore",
|
||||||
".*Converting a tensor to a Python boolean might cause the trace to be incorrect.*",
|
".*Converting a tensor to a Python boolean might cause the trace to be incorrect.*",
|
||||||
)
|
)
|
||||||
|
warnings.filterwarnings(
|
||||||
|
"ignore",
|
||||||
|
".*Overwriting tiny_vit_.*",
|
||||||
|
)
|
||||||
|
|
||||||
logger = getLogger(__name__)
|
logger = getLogger(__name__)
|
||||||
|
|
||||||
|
|
|
@ -1,20 +1,29 @@
|
||||||
# https://github.com/ForserX/StableDiffusionUI/blob/main/data/repo/diffusion_scripts/controlnet_pipe.py
|
# https://github.com/ForserX/StableDiffusionUI/blob/main/data/repo/diffusion_scripts/controlnet_pipe.py
|
||||||
|
|
||||||
from logging import getLogger
|
import warnings
|
||||||
from os import path
|
|
||||||
|
|
||||||
import cv2
|
warnings.filterwarnings(
|
||||||
import numpy as np
|
"ignore",
|
||||||
import torch
|
".*Overwriting tiny_vit_.*",
|
||||||
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 https://stackoverflow.com/a/65418681
|
||||||
from .ade_palette import ade_palette
|
if True: # noqa: E402
|
||||||
from .laion_face import generate_annotation
|
from logging import getLogger
|
||||||
from .noise_source import noise_source_histogram
|
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__)
|
logger = getLogger(__name__)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue