1
0
Fork 0

lint(api): remove some confusing log messages

This commit is contained in:
Sean Sube 2023-12-23 12:17:23 -06:00
parent 972f9de2ef
commit 52065ef317
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
4 changed files with 0 additions and 6 deletions

View File

@ -27,7 +27,6 @@ def convert_correction_gfpgan(
logger.info("ONNX model already exists, skipping")
return
logger.info("loading and training model")
model = RRDBNet(
num_in_ch=3,
num_out_ch=3,
@ -38,7 +37,6 @@ def convert_correction_gfpgan(
)
torch_model = torch.load(source, map_location=conversion.map_location)
# TODO: make sure strict=False is safe here
if "params_ema" in torch_model:
model.load_state_dict(torch_model["params_ema"], strict=False)
else:

View File

@ -27,7 +27,6 @@ def convert_upscaling_bsrgan(
logger.info("ONNX model already exists, skipping")
return
logger.info("loading and training model")
# values based on https://github.com/cszn/BSRGAN/blob/main/main_test_bsrgan.py#L69
model = RRDBNet(
num_in_ch=3,

View File

@ -31,8 +31,6 @@ def convert_upscale_resrgan(
logger.info("ONNX model already exists, skipping")
return
logger.info("loading and training model")
if TAG_X4_V3 in name:
# the x4-v3 model needs a different network
model = SRVGGNetCompact(

View File

@ -27,7 +27,6 @@ def convert_upscaling_swinir(
logger.info("ONNX model already exists, skipping")
return
logger.info("loading and training model")
# values based on https://github.com/JingyunLiang/SwinIR/blob/main/main_test_swinir.py#L128
params = {
"depths": [6] * 6,