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 get_latents_from_seed, load_pipeline, optimize_pipeline
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 .image import (
    expand_image,
    mask_filter_gaussian_multiply,
    mask_filter_gaussian_screen,
    mask_filter_none,
    noise_source_fill_edge,
    noise_source_fill_mask,
    noise_source_gaussian,
    noise_source_histogram,
    noise_source_normal,
    noise_source_uniform,
    valid_image,
)
from .onnx import OnnxNet, 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 .upscale import run_upscale_correction
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,
)