77 lines
1.8 KiB
Python
77 lines
1.8 KiB
Python
from . import logging
|
|
from .chain import (
|
|
correct_codeformer,
|
|
correct_gfpgan,
|
|
upscale_resrgan,
|
|
upscale_stable_diffusion,
|
|
)
|
|
from .convert.diffusion.lora import blend_loras
|
|
from .convert.diffusion.textual_inversion import blend_textual_inversions
|
|
from .diffusers.load import load_pipeline, optimize_pipeline
|
|
from .diffusers.utils import get_tile_latents, get_latents_from_seed
|
|
from .diffusers.run import (
|
|
run_blend_pipeline,
|
|
run_img2img_pipeline,
|
|
run_inpaint_pipeline,
|
|
run_txt2img_pipeline,
|
|
run_upscale_pipeline,
|
|
)
|
|
from .diffusers.stub_scheduler import StubScheduler
|
|
from .diffusers.upscale import stage_upscale_correction
|
|
from .image.utils import (
|
|
expand_image,
|
|
)
|
|
from .image.mask_filter import (
|
|
mask_filter_gaussian_multiply,
|
|
mask_filter_gaussian_screen,
|
|
mask_filter_none,
|
|
)
|
|
from .image.noise_source import (
|
|
noise_source_fill_edge,
|
|
noise_source_fill_mask,
|
|
noise_source_gaussian,
|
|
noise_source_histogram,
|
|
noise_source_normal,
|
|
noise_source_uniform,
|
|
)
|
|
from .image.source_filter import (
|
|
source_filter_canny,
|
|
source_filter_depth,
|
|
source_filter_hed,
|
|
source_filter_mlsd,
|
|
source_filter_normal,
|
|
source_filter_openpose,
|
|
source_filter_scribble,
|
|
source_filter_segment,
|
|
)
|
|
from .onnx import OnnxRRDBNet, OnnxTensor
|
|
from .params import (
|
|
Border,
|
|
DeviceParams,
|
|
ImageParams,
|
|
Param,
|
|
Point,
|
|
Size,
|
|
StageParams,
|
|
UpscaleParams,
|
|
)
|
|
from .server import (
|
|
ModelCache,
|
|
ServerContext,
|
|
apply_patch_basicsr,
|
|
apply_patch_codeformer,
|
|
apply_patch_facexlib,
|
|
apply_patches,
|
|
)
|
|
from .utils import (
|
|
base_join,
|
|
get_and_clamp_float,
|
|
get_and_clamp_int,
|
|
get_from_list,
|
|
get_from_map,
|
|
get_not_empty,
|
|
)
|
|
from .worker import (
|
|
DevicePoolExecutor,
|
|
)
|