fix(api): switch between available pipelines
This commit is contained in:
parent
317029356e
commit
b20df68131
|
@ -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(),
|
||||
|
|
|
@ -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(),
|
||||
|
|
|
@ -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(),
|
||||
|
|
|
@ -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,
|
||||
)
|
||||
|
||||
|
|
|
@ -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(),
|
||||
|
|
Loading…
Reference in New Issue