onnx-web/api/onnx_web/__init__.py

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 run_upscale_correction
from .image.utils import (
    expand_image,
    valid_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,
)