fix some logs
This commit is contained in:
parent
8c54de7a52
commit
9aad798a4a
|
@ -105,7 +105,7 @@ class UpscaleRealESRGANStage(BaseStage):
|
||||||
stage_source: Optional[Image.Image] = None,
|
stage_source: Optional[Image.Image] = None,
|
||||||
**kwargs,
|
**kwargs,
|
||||||
) -> StageResult:
|
) -> StageResult:
|
||||||
logger.info("upscaling image with Real ESRGAN: x%s", upscale.scale)
|
logger.info("upscaling with ESRGAN model: %s", upscale.upscale_model)
|
||||||
|
|
||||||
upsampler = self.load(
|
upsampler = self.load(
|
||||||
server, upscale, worker.get_device(), tile=stage.tile_size
|
server, upscale, worker.get_device(), tile=stage.tile_size
|
||||||
|
|
|
@ -67,7 +67,7 @@ class UpscaleSwinIRStage(BaseStage):
|
||||||
logger.warning("no correction model given, skipping")
|
logger.warning("no correction model given, skipping")
|
||||||
return sources
|
return sources
|
||||||
|
|
||||||
logger.info("correcting faces with SwinIR model: %s", upscale.upscale_model)
|
logger.info("upscaling with SwinIR model: %s", upscale.upscale_model)
|
||||||
device = worker.get_device()
|
device = worker.get_device()
|
||||||
swinir = self.load(server, stage, upscale, device)
|
swinir = self.load(server, stage, upscale, device)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue