1
0
Fork 0

fix(api): switch between available pipelines

This commit is contained in:
Sean Sube 2023-04-15 15:37:45 -05:00
parent 317029356e
commit b20df68131
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
5 changed files with 18 additions and 12 deletions

View File

@ -30,9 +30,10 @@ def blend_img2img(
"blending image using img2img, %s steps: %s", params.steps, params.prompt
)
pipe_type = "lpw" if params.lpw() else "img2img"
pipe = load_pipeline(
server,
"img2img",
pipe_type,
params.model,
params.scheduler,
job.get_device(),

View File

@ -58,9 +58,10 @@ def blend_inpaint(
save_image(server, "last-mask.png", stage_mask)
save_image(server, "last-noise.png", noise)
pipe_type = "lpw" if params.lpw() else "inpaint"
pipe = load_pipeline(
server,
"inpaint",
pipe_type,
params.model,
params.scheduler,
job.get_device(),

View File

@ -36,9 +36,10 @@ def source_txt2img(
)
latents = get_latents_from_seed(params.seed, size)
pipe_type = "lpw" if params.lpw() else "txt2img"
pipe = load_pipeline(
server,
"txt2img",
pipe_type,
params.model,
params.scheduler,
job.get_device(),

View File

@ -72,9 +72,10 @@ def upscale_outpaint(
save_image(server, "tile-mask.png", tile_mask)
latents = get_tile_latents(full_latents, dims)
pipe_type = "lpw" if params.lpw() else "inpaint"
pipe = load_pipeline(
server,
"inpaint",
pipe_type,
params.model,
params.scheduler,
job.get_device(),
@ -100,14 +101,14 @@ def upscale_outpaint(
result = pipe(
params.prompt,
tile_source,
generator=rng,
guidance_scale=params.cfg,
tile_mask,
height=size.height,
latents=latents,
mask_image=tile_mask,
negative_prompt=params.negative_prompt,
num_inference_steps=params.steps,
width=size.width,
num_inference_steps=params.steps,
guidance_scale=params.cfg,
negative_prompt=params.negative_prompt,
generator=rng,
latents=latents,
callback=callback,
)

View File

@ -65,9 +65,10 @@ def run_highres(
)
# load img2img pipeline once
pipe_type = "lpw" if params.lpw() else "img2img"
highres_pipe = load_pipeline(
server,
"img2img",
pipe_type,
params.model,
params.scheduler,
job.get_device(),
@ -168,9 +169,10 @@ def run_txt2img_pipeline(
(prompt, inversions) = get_inversions_from_prompt(prompt)
params.prompt = prompt
pipe_type = "lpw" if params.lpw() else "txt2img"
pipe = load_pipeline(
server,
"txt2img",
pipe_type,
params.model,
params.scheduler,
job.get_device(),