From 8d92dac82946d68881c30a5d648fbfb97e8490b3 Mon Sep 17 00:00:00 2001 From: mrq Date: Sun, 9 Jun 2024 11:12:30 -0500 Subject: [PATCH] forgot I renamed this --- vall_e/engines/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vall_e/engines/__init__.py b/vall_e/engines/__init__.py index 8b933f4..97bd9fc 100755 --- a/vall_e/engines/__init__.py +++ b/vall_e/engines/__init__.py @@ -147,8 +147,8 @@ def load_engines(training=True): state["text_emb.weight"] = state["text_emb.weight"][:model.config.text_tokens] # resize text embedding - if "rvq_level_emb.weight" in state and model.config.resp_levels != state["rvq_level_emb.weight"].shape[0]: - state["rvq_level_emb.weight"] = state["rvq_level_emb.weight"][:model.config.resp_levels] + if "rvq_l_emb.weight" in state and model.config.resp_levels != state["rvq_l_emb.weight"].shape[0]: + state["rvq_l_emb.weight"] = state["rvq_l_emb.weight"][:model.config.resp_levels] model.load_state_dict(state, strict=cfg.trainer.strict_loading)