feat(api): enable 1x upscaling models
This commit is contained in:
parent
11e643bcb5
commit
7abe6dc6a9
|
@ -68,7 +68,6 @@ def stage_upscale_correction(
|
||||||
"upscale": upscale,
|
"upscale": upscale,
|
||||||
}
|
}
|
||||||
upscale_stage: Optional[PipelineStage] = None
|
upscale_stage: Optional[PipelineStage] = None
|
||||||
if upscale.scale > 1:
|
|
||||||
if "bsrgan" in upscale.upscale_model:
|
if "bsrgan" in upscale.upscale_model:
|
||||||
bsrgan_params = StageParams(
|
bsrgan_params = StageParams(
|
||||||
tile_size=stage.tile_size,
|
tile_size=stage.tile_size,
|
||||||
|
|
Loading…
Reference in New Issue