apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix some logs

This commit is contained in:
Sean Sube 2023-12-31 11:28:57 -06:00
parent 8c54de7a52
commit 9aad798a4a
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/chain/upscale_resrgan.py
View File

@ -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

2
api/onnx_web/chain/upscale_swinir.py
View File

@ -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)