From 9aad798a4ac6b2a35c451020043423684e956364 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 31 Dec 2023 11:28:57 -0600 Subject: [PATCH] fix some logs --- api/onnx_web/chain/upscale_resrgan.py | 2 +- api/onnx_web/chain/upscale_swinir.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/chain/upscale_resrgan.py b/api/onnx_web/chain/upscale_resrgan.py index 33338b43..b0370e7b 100644 --- a/api/onnx_web/chain/upscale_resrgan.py +++ b/api/onnx_web/chain/upscale_resrgan.py @@ -105,7 +105,7 @@ class UpscaleRealESRGANStage(BaseStage): stage_source: Optional[Image.Image] = None, **kwargs, ) -> 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( server, upscale, worker.get_device(), tile=stage.tile_size diff --git a/api/onnx_web/chain/upscale_swinir.py b/api/onnx_web/chain/upscale_swinir.py index ef7d421f..aa677e3d 100644 --- a/api/onnx_web/chain/upscale_swinir.py +++ b/api/onnx_web/chain/upscale_swinir.py @@ -67,7 +67,7 @@ class UpscaleSwinIRStage(BaseStage): logger.warning("no correction model given, skipping") 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() swinir = self.load(server, stage, upscale, device)