mirror of
https://github.com/Stability-AI/generative-models.git
synced 2026-02-02 04:14:27 +01:00
Use wrapper correctly in refiner helper
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user