fix(api): make sure each upscale/correction model gets its own stage
This commit is contained in:
parent
de4e7b0dc9
commit
833fc5c2f8
|
@ -35,21 +35,21 @@ def run_upscale_correction(
|
||||||
|
|
||||||
if upscale.scale > 1:
|
if upscale.scale > 1:
|
||||||
if "esrgan" in upscale.upscale_model:
|
if "esrgan" in upscale.upscale_model:
|
||||||
stage = StageParams(tile_size=stage.tile_size, outscale=upscale.outscale)
|
resr_stage = StageParams(tile_size=stage.tile_size, outscale=upscale.outscale)
|
||||||
chain.append((upscale_resrgan, stage, None))
|
chain.append((upscale_resrgan, resr_stage, None))
|
||||||
elif "stable-diffusion" in upscale.upscale_model:
|
elif "stable-diffusion" in upscale.upscale_model:
|
||||||
mini_tile = min(SizeChart.mini, stage.tile_size)
|
mini_tile = min(SizeChart.mini, stage.tile_size)
|
||||||
stage = StageParams(tile_size=mini_tile, outscale=upscale.outscale)
|
sd_stage = StageParams(tile_size=mini_tile, outscale=upscale.outscale)
|
||||||
chain.append((upscale_stable_diffusion, stage, None))
|
chain.append((upscale_stable_diffusion, sd_stage, None))
|
||||||
else:
|
else:
|
||||||
logger.warn("unknown upscaling model: %s", upscale.upscale_model)
|
logger.warn("unknown upscaling model: %s", upscale.upscale_model)
|
||||||
|
|
||||||
if upscale.faces:
|
if upscale.faces:
|
||||||
stage = StageParams(tile_size=stage.tile_size, outscale=1)
|
face_stage = StageParams(tile_size=stage.tile_size, outscale=upscale.face_outscale)
|
||||||
if "codeformer" in upscale.correction_model:
|
if "codeformer" in upscale.correction_model:
|
||||||
chain.append((correct_codeformer, stage, None))
|
chain.append((correct_codeformer, face_stage, None))
|
||||||
elif "gfpgan" in upscale.correction_model:
|
elif "gfpgan" in upscale.correction_model:
|
||||||
chain.append((correct_gfpgan, stage, None))
|
chain.append((correct_gfpgan, face_stage, None))
|
||||||
else:
|
else:
|
||||||
logger.warn("unknown correction model: %s", upscale.correction_model)
|
logger.warn("unknown correction model: %s", upscale.correction_model)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue