Use wrapper correctly in refiner helper

This commit is contained in:
Stephan Auerhahn
2023-08-02 23:14:30 +00:00
parent 45c443b316
commit 45feb6cb9c

View File

@@ -268,6 +268,12 @@ class SamplingPipeline:
"negative_aesthetic_score": 2.5,
}
if params.img2img_strength < 1.0:
sampler.discretization = Img2ImgDiscretizationWrapper(
sampler.discretization,
strength=params.img2img_strength,
)
return do_img2img(
image,
self.model,