1
0
Fork 0

fix(api): make img2img blend work with LPW

This commit is contained in:
Sean Sube 2023-07-03 08:05:30 -05:00
parent 12698d38eb
commit 197dc48b06
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions

View File

@ -61,10 +61,10 @@ class BlendImg2ImgStage(BaseStage):
logger.debug("using LPW pipeline for img2img")
rng = torch.manual_seed(params.seed)
result = pipe.img2img(
source,
params.prompt,
generator=rng,
guidance_scale=params.cfg,
image=source,
negative_prompt=params.negative_prompt,
num_inference_steps=params.steps,
callback=callback,