chore(api): add logging for LPW pipelines
This commit is contained in:
parent
4abd0cb8a3
commit
20467aafac
|
@ -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,
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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,
|
||||
|
|
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue