fix imports, add missing controlnet deps
This commit is contained in:
parent
4df28a5ce7
commit
428fb06ecf
|
@ -40,7 +40,7 @@ from .image.source_filter import (
|
||||||
source_filter_hed,
|
source_filter_hed,
|
||||||
source_filter_mlsd,
|
source_filter_mlsd,
|
||||||
source_filter_normal,
|
source_filter_normal,
|
||||||
source_filter_pose,
|
source_filter_openpose,
|
||||||
source_filter_scribble,
|
source_filter_scribble,
|
||||||
source_filter_segment,
|
source_filter_segment,
|
||||||
)
|
)
|
||||||
|
|
|
@ -231,7 +231,9 @@ def run_img2img_pipeline(
|
||||||
|
|
||||||
# filter the source image
|
# filter the source image
|
||||||
if source_filter is not None:
|
if source_filter is not None:
|
||||||
source = get_source_filters(source_filter)(source)
|
f = get_source_filters().get(source_filter, None)
|
||||||
|
if f is not None:
|
||||||
|
source = f(server, source)
|
||||||
|
|
||||||
pipe = load_pipeline(
|
pipe = load_pipeline(
|
||||||
server,
|
server,
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
from PIL import Image, ImageChops, ImageFilter
|
from PIL import Image, ImageChops, ImageFilter
|
||||||
|
|
||||||
from .params import Point
|
from ..params import Point
|
||||||
|
|
||||||
|
|
||||||
def mask_filter_none(
|
def mask_filter_none(
|
||||||
|
|
|
@ -2,7 +2,7 @@ import numpy as np
|
||||||
from numpy import random
|
from numpy import random
|
||||||
from PIL import Image, ImageFilter
|
from PIL import Image, ImageFilter
|
||||||
|
|
||||||
from .params import Point
|
from ..params import Point
|
||||||
|
|
||||||
|
|
||||||
def get_pixel_index(x: int, y: int, width: int) -> int:
|
def get_pixel_index(x: int, y: int, width: int) -> int:
|
||||||
|
|
|
@ -4,7 +4,7 @@ from PIL import Image, ImageChops, ImageOps
|
||||||
|
|
||||||
from .mask_filter import mask_filter_none
|
from .mask_filter import mask_filter_none
|
||||||
from .noise_source import noise_source_histogram
|
from .noise_source import noise_source_histogram
|
||||||
from .params import Border, Size
|
from ..params import Border, Size
|
||||||
|
|
||||||
|
|
||||||
# very loosely based on https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/master/scripts/outpainting_mk_2.py#L175-L232
|
# very loosely based on https://github.com/AUTOMATIC1111/stable-diffusion-webui/blob/master/scripts/outpainting_mk_2.py#L175-L232
|
||||||
|
|
|
@ -7,6 +7,7 @@ accelerate
|
||||||
coloredlogs
|
coloredlogs
|
||||||
controlnet_aux
|
controlnet_aux
|
||||||
diffusers
|
diffusers
|
||||||
|
mediapipe
|
||||||
onnx
|
onnx
|
||||||
# onnxruntime has many platform-specific packages
|
# onnxruntime has many platform-specific packages
|
||||||
safetensors
|
safetensors
|
||||||
|
|
Loading…
Reference in New Issue