From 0ae82e60b487e3d57f894e8767d356fb3fac3d8c Mon Sep 17 00:00:00 2001 From: "Salvador E. Tropea" Date: Sun, 6 Jul 2025 15:07:40 -0300 Subject: [PATCH] [Load Model][Added] Support for other models Just avoid accessing data particular for a model --- source/db/load_model.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/source/db/load_model.py b/source/db/load_model.py index 9f2f9e0..273b57f 100644 --- a/source/db/load_model.py +++ b/source/db/load_model.py @@ -15,10 +15,12 @@ logger = logging.getLogger(f"{NODES_NAME}.load_model") def show_model_parameters(d): - logger.debug("Using model parameters:") - logger.debug(f" Frequency Dimension (dim_f): {d['mdx_dim_f_set']}") - logger.debug(f" Base Channels (ch): {d['channels']}") - logger.debug(f" U-Net Stages: {d['stages']}") + model_t = d['model_t'].lower() + if model_t == 'mdx': + logger.debug("Using model parameters:") + logger.debug(f" Frequency Dimension (dim_f): {d['mdx_dim_f_set']}") + logger.debug(f" Base Channels (ch): {d['channels']}") + logger.debug(f" U-Net Stages: {d['stages']}") def load_model(d, device, models_dir):