nodes/set-soft-AudioSeparation
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Load Model][Added] Support for other models

Browse Source 
Just avoid accessing data particular for a model
This commit is contained in:
Salvador E. Tropea
2025-07-06 15:07:40 -03:00
parent f1841269a0
commit 0ae82e60b4
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/db/load_model.py
+6 -4
View File
@@ -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):