2023-02-05 13:53:26 +00:00
|
|
|
from .base import ChainPipeline, PipelineStage, StageCallback, StageParams
|
2023-04-12 00:29:25 +00:00
|
|
|
from .blend_controlnet import blend_controlnet
|
2023-02-05 13:53:26 +00:00
|
|
|
from .blend_img2img import blend_img2img
|
|
|
|
from .blend_inpaint import blend_inpaint
|
2023-04-11 02:15:57 +00:00
|
|
|
from .blend_linear import blend_linear
|
2023-02-13 23:34:42 +00:00
|
|
|
from .blend_mask import blend_mask
|
2023-04-12 00:29:25 +00:00
|
|
|
from .blend_pix2pix import blend_pix2pix
|
2023-02-05 14:37:47 +00:00
|
|
|
from .correct_codeformer import correct_codeformer
|
2023-02-05 13:53:26 +00:00
|
|
|
from .correct_gfpgan import correct_gfpgan
|
|
|
|
from .persist_disk import persist_disk
|
|
|
|
from .persist_s3 import persist_s3
|
|
|
|
from .reduce_crop import reduce_crop
|
|
|
|
from .reduce_thumbnail import reduce_thumbnail
|
|
|
|
from .source_noise import source_noise
|
2023-04-11 02:03:19 +00:00
|
|
|
from .source_s3 import source_s3
|
2023-02-05 13:53:26 +00:00
|
|
|
from .source_txt2img import source_txt2img
|
2023-04-11 02:03:19 +00:00
|
|
|
from .source_url import source_url
|
2023-04-10 22:49:56 +00:00
|
|
|
from .upscale_bsrgan import upscale_bsrgan
|
2023-02-05 13:53:26 +00:00
|
|
|
from .upscale_outpaint import upscale_outpaint
|
|
|
|
from .upscale_resrgan import upscale_resrgan
|
|
|
|
from .upscale_stable_diffusion import upscale_stable_diffusion
|
2023-04-10 22:49:56 +00:00
|
|
|
from .upscale_swinir import upscale_swinir
|
2023-02-26 16:15:12 +00:00
|
|
|
|
|
|
|
CHAIN_STAGES = {
|
2023-04-12 00:29:25 +00:00
|
|
|
"blend-controlnet": blend_controlnet,
|
2023-02-26 16:15:12 +00:00
|
|
|
"blend-img2img": blend_img2img,
|
|
|
|
"blend-inpaint": blend_inpaint,
|
2023-04-11 02:15:57 +00:00
|
|
|
"blend-linear": blend_linear,
|
2023-02-26 16:15:12 +00:00
|
|
|
"blend-mask": blend_mask,
|
2023-04-12 00:29:25 +00:00
|
|
|
"blend-pix2pix": blend_pix2pix,
|
2023-02-26 16:15:12 +00:00
|
|
|
"correct-codeformer": correct_codeformer,
|
|
|
|
"correct-gfpgan": correct_gfpgan,
|
|
|
|
"persist-disk": persist_disk,
|
|
|
|
"persist-s3": persist_s3,
|
|
|
|
"reduce-crop": reduce_crop,
|
|
|
|
"reduce-thumbnail": reduce_thumbnail,
|
|
|
|
"source-noise": source_noise,
|
2023-04-11 02:03:19 +00:00
|
|
|
"source-s3": source_s3,
|
2023-02-26 16:15:12 +00:00
|
|
|
"source-txt2img": source_txt2img,
|
2023-04-11 02:03:19 +00:00
|
|
|
"source-url": source_url,
|
2023-04-10 22:49:56 +00:00
|
|
|
"upscale-bsrgan": upscale_bsrgan,
|
2023-02-26 16:15:12 +00:00
|
|
|
"upscale-outpaint": upscale_outpaint,
|
|
|
|
"upscale-resrgan": upscale_resrgan,
|
|
|
|
"upscale-stable-diffusion": upscale_stable_diffusion,
|
2023-04-10 22:49:56 +00:00
|
|
|
"upscale-swinir": upscale_swinir,
|
2023-02-26 20:15:30 +00:00
|
|
|
}
|