2023-01-28 23:15:42 +00:00
|
|
|
from . import logging
|
2023-02-26 16:15:12 +00:00
|
|
|
from .chain import (
|
|
|
|
correct_codeformer,
|
|
|
|
correct_gfpgan,
|
|
|
|
upscale_resrgan,
|
|
|
|
upscale_stable_diffusion,
|
|
|
|
)
|
2023-02-19 00:55:42 +00:00
|
|
|
from .diffusion.load import get_latents_from_seed, load_pipeline, optimize_pipeline
|
2023-01-29 02:15:39 +00:00
|
|
|
from .diffusion.run import (
|
2023-02-19 00:55:42 +00:00
|
|
|
run_blend_pipeline,
|
2023-02-05 13:53:26 +00:00
|
|
|
run_img2img_pipeline,
|
|
|
|
run_inpaint_pipeline,
|
|
|
|
run_txt2img_pipeline,
|
2023-02-19 00:55:42 +00:00
|
|
|
run_upscale_pipeline,
|
2023-01-26 02:31:39 +00:00
|
|
|
)
|
2023-02-19 00:55:42 +00:00
|
|
|
from .diffusion.stub_scheduler import StubScheduler
|
2023-01-16 01:47:57 +00:00
|
|
|
from .image import (
|
2023-02-05 13:53:26 +00:00
|
|
|
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,
|
2023-02-19 00:55:42 +00:00
|
|
|
valid_image,
|
|
|
|
)
|
|
|
|
from .onnx import OnnxNet, OnnxTensor
|
|
|
|
from .params import (
|
|
|
|
Border,
|
2023-02-26 05:49:39 +00:00
|
|
|
DeviceParams,
|
2023-02-19 00:55:42 +00:00
|
|
|
ImageParams,
|
|
|
|
Param,
|
|
|
|
Point,
|
|
|
|
Size,
|
|
|
|
StageParams,
|
|
|
|
UpscaleParams,
|
|
|
|
)
|
|
|
|
from .server import (
|
|
|
|
ModelCache,
|
2023-02-19 02:28:21 +00:00
|
|
|
ServerContext,
|
2023-02-19 00:55:42 +00:00
|
|
|
apply_patch_basicsr,
|
|
|
|
apply_patch_codeformer,
|
|
|
|
apply_patch_facexlib,
|
|
|
|
apply_patches,
|
2023-01-16 01:47:57 +00:00
|
|
|
)
|
2023-02-19 02:28:21 +00:00
|
|
|
from .upscale import run_upscale_correction
|
2023-01-16 01:34:51 +00:00
|
|
|
from .utils import (
|
2023-02-05 13:53:26 +00:00
|
|
|
base_join,
|
|
|
|
get_and_clamp_float,
|
|
|
|
get_and_clamp_int,
|
|
|
|
get_from_list,
|
|
|
|
get_from_map,
|
|
|
|
get_not_empty,
|
|
|
|
)
|
2023-02-26 05:49:39 +00:00
|
|
|
from .worker import (
|
|
|
|
DevicePoolExecutor,
|
2023-02-26 20:15:30 +00:00
|
|
|
)
|