diff --git a/api/onnx_web/chain/blend_img2img.py b/api/onnx_web/chain/blend_img2img.py index 91ac3da9..8fed3a40 100644 --- a/api/onnx_web/chain/blend_img2img.py +++ b/api/onnx_web/chain/blend_img2img.py @@ -35,6 +35,7 @@ def blend_img2img( params.lpw, ) if params.lpw: + logger.debug('using LPW pipeline for img2img') rng = torch.manual_seed(params.seed) result = pipe.img2img( prompt, diff --git a/api/onnx_web/chain/blend_inpaint.py b/api/onnx_web/chain/blend_inpaint.py index cc9f25ba..641f9c0c 100644 --- a/api/onnx_web/chain/blend_inpaint.py +++ b/api/onnx_web/chain/blend_inpaint.py @@ -70,6 +70,7 @@ def blend_inpaint( ) if params.lpw: + logger.debug('using LPW pipeline for inpaint') rng = torch.manual_seed(params.seed) result = pipe.inpaint( params.prompt, diff --git a/api/onnx_web/chain/source_txt2img.py b/api/onnx_web/chain/source_txt2img.py index 3edb9365..c565d079 100644 --- a/api/onnx_web/chain/source_txt2img.py +++ b/api/onnx_web/chain/source_txt2img.py @@ -38,6 +38,7 @@ def source_txt2img( ) if params.lpw: + logger.debug('using LPW pipeline for txt2img') rng = torch.manual_seed(params.seed) result = pipe.text2img( prompt, diff --git a/api/onnx_web/chain/upscale_outpaint.py b/api/onnx_web/chain/upscale_outpaint.py index 8c7e8896..ea93ff88 100644 --- a/api/onnx_web/chain/upscale_outpaint.py +++ b/api/onnx_web/chain/upscale_outpaint.py @@ -75,6 +75,7 @@ def upscale_outpaint( params.lpw, ) if params.lpw: + logger.debug('using LPW pipeline for inpaint') rng = torch.manual_seed(params.seed) result = pipe.inpaint( image, diff --git a/api/onnx_web/diffusion/run.py b/api/onnx_web/diffusion/run.py index 4ab79de0..ebf6b429 100644 --- a/api/onnx_web/diffusion/run.py +++ b/api/onnx_web/diffusion/run.py @@ -32,6 +32,7 @@ def run_txt2img_pipeline( progress = job.get_progress_callback() if params.lpw: + logger.debug('using LPW pipeline for txt2img') rng = torch.manual_seed(params.seed) result = pipe.text2img( params.prompt, @@ -92,6 +93,7 @@ def run_img2img_pipeline( ) progress = job.get_progress_callback() if params.lpw: + logger.debug('using LPW pipeline for img2img') rng = torch.manual_seed(params.seed) result = pipe.img2img( source_image,