apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

chore(api): add logging for LPW pipelines

This commit is contained in:
Sean Sube 2023-02-05 17:53:51 -06:00
parent 4abd0cb8a3
commit 20467aafac
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
5 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/onnx_web/chain/blend_img2img.py
View File

@ -35,6 +35,7 @@ def blend_img2img(
params.lpw, params.lpw,
) )
if params.lpw: if params.lpw:
logger.debug('using LPW pipeline for img2img')
rng = torch.manual_seed(params.seed) rng = torch.manual_seed(params.seed)
result = pipe.img2img( result = pipe.img2img(
prompt, prompt,

1
api/onnx_web/chain/blend_inpaint.py
View File

@ -70,6 +70,7 @@ def blend_inpaint(
) )
if params.lpw: if params.lpw:
logger.debug('using LPW pipeline for inpaint')
rng = torch.manual_seed(params.seed) rng = torch.manual_seed(params.seed)
result = pipe.inpaint( result = pipe.inpaint(
params.prompt, params.prompt,

1
api/onnx_web/chain/source_txt2img.py
View File

@ -38,6 +38,7 @@ def source_txt2img(
) )
if params.lpw: if params.lpw:
logger.debug('using LPW pipeline for txt2img')
rng = torch.manual_seed(params.seed) rng = torch.manual_seed(params.seed)
result = pipe.text2img( result = pipe.text2img(
prompt, prompt,

1
api/onnx_web/chain/upscale_outpaint.py
View File

@ -75,6 +75,7 @@ def upscale_outpaint(
params.lpw, params.lpw,
) )
if params.lpw: if params.lpw:
logger.debug('using LPW pipeline for inpaint')
rng = torch.manual_seed(params.seed) rng = torch.manual_seed(params.seed)
result = pipe.inpaint( result = pipe.inpaint(
image, image,

2
api/onnx_web/diffusion/run.py
View File

@ -32,6 +32,7 @@ def run_txt2img_pipeline(
progress = job.get_progress_callback() progress = job.get_progress_callback()
if params.lpw: if params.lpw:
logger.debug('using LPW pipeline for txt2img')
rng = torch.manual_seed(params.seed) rng = torch.manual_seed(params.seed)
result = pipe.text2img( result = pipe.text2img(
params.prompt, params.prompt,
@ -92,6 +93,7 @@ def run_img2img_pipeline(
) )
progress = job.get_progress_callback() progress = job.get_progress_callback()
if params.lpw: if params.lpw:
logger.debug('using LPW pipeline for img2img')
rng = torch.manual_seed(params.seed) rng = torch.manual_seed(params.seed)
result = pipe.img2img( result = pipe.img2img(
source_image, source_image,