onnx-web/api/onnx_web/__init__.py

from . import logging
from .chain import correct_gfpgan, upscale_resrgan, upscale_stable_diffusion
from .diffusion.load import get_latents_from_seed, load_pipeline, optimize_pipeline
from .diffusion.run import (
    run_blend_pipeline,
    run_img2img_pipeline,
    run_inpaint_pipeline,
    run_txt2img_pipeline,
    run_upscale_pipeline,
)
from .diffusion.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,
    ImageParams,
    Param,
    Point,
    Size,
    StageParams,
    UpscaleParams,
)
from .server import (
    DeviceParams,
    DevicePoolExecutor,
    ModelCache,
    apply_patch_basicsr,
    apply_patch_codeformer,
    apply_patch_facexlib,
    apply_patches,
    run_upscale_correction,
)
from .utils import (
    ServerContext,
    base_join,
    get_and_clamp_float,
    get_and_clamp_int,
    get_from_list,
    get_from_map,
    get_not_empty,
)