feat(api): enable LPW custom pipeline (#27)
This commit is contained in:
parent
1e477f7548
commit
7b506cb6d3
|
@ -75,14 +75,14 @@ def upscale_outpaint(
|
||||||
latents = get_tile_latents(full_latents, dims)
|
latents = get_tile_latents(full_latents, dims)
|
||||||
rng = np.random.RandomState(params.seed)
|
rng = np.random.RandomState(params.seed)
|
||||||
|
|
||||||
result = pipe(
|
result = pipe.inpaint(
|
||||||
|
image,
|
||||||
|
mask,
|
||||||
prompt,
|
prompt,
|
||||||
generator=rng,
|
generator=rng,
|
||||||
guidance_scale=params.cfg,
|
guidance_scale=params.cfg,
|
||||||
height=size.height,
|
height=size.height,
|
||||||
image=image,
|
|
||||||
latents=latents,
|
latents=latents,
|
||||||
mask_image=mask,
|
|
||||||
negative_prompt=params.negative_prompt,
|
negative_prompt=params.negative_prompt,
|
||||||
num_inference_steps=params.steps,
|
num_inference_steps=params.steps,
|
||||||
width=size.width,
|
width=size.width,
|
||||||
|
|
|
@ -73,6 +73,8 @@ def load_pipeline(
|
||||||
model,
|
model,
|
||||||
provider=device.provider,
|
provider=device.provider,
|
||||||
provider_options=device.options,
|
provider_options=device.options,
|
||||||
|
custom_pipeline='lpw_stable_diffusion_onnx',
|
||||||
|
revision='onnx',
|
||||||
safety_checker=None,
|
safety_checker=None,
|
||||||
scheduler=scheduler,
|
scheduler=scheduler,
|
||||||
)
|
)
|
||||||
|
|
|
@ -32,7 +32,7 @@ def run_txt2img_pipeline(
|
||||||
rng = np.random.RandomState(params.seed)
|
rng = np.random.RandomState(params.seed)
|
||||||
|
|
||||||
progress = job.get_progress_callback()
|
progress = job.get_progress_callback()
|
||||||
result = pipe(
|
result = pipe.txt2img(
|
||||||
params.prompt,
|
params.prompt,
|
||||||
height=size.height,
|
height=size.height,
|
||||||
width=size.width,
|
width=size.width,
|
||||||
|
@ -77,11 +77,11 @@ def run_img2img_pipeline(
|
||||||
rng = np.random.RandomState(params.seed)
|
rng = np.random.RandomState(params.seed)
|
||||||
|
|
||||||
progress = job.get_progress_callback()
|
progress = job.get_progress_callback()
|
||||||
result = pipe(
|
result = pipe.img2img(
|
||||||
|
source_image,
|
||||||
params.prompt,
|
params.prompt,
|
||||||
generator=rng,
|
generator=rng,
|
||||||
guidance_scale=params.cfg,
|
guidance_scale=params.cfg,
|
||||||
image=source_image,
|
|
||||||
negative_prompt=params.negative_prompt,
|
negative_prompt=params.negative_prompt,
|
||||||
num_inference_steps=params.steps,
|
num_inference_steps=params.steps,
|
||||||
strength=strength,
|
strength=strength,
|
||||||
|
|
Loading…
Reference in New Issue