From e513d2ef1925f670d424633682dca757a6f9dc74 Mon Sep 17 00:00:00 2001 From: mrq Date: Sat, 23 Dec 2023 16:08:17 -0600 Subject: [PATCH] experts weren't forwarded into constructer (wasted a few days of training garbage) --- vall_e/models/__init__.py | 1 + 1 file changed, 1 insertion(+) diff --git a/vall_e/models/__init__.py b/vall_e/models/__init__.py index 38071e1..1fadf2c 100755 --- a/vall_e/models/__init__.py +++ b/vall_e/models/__init__.py @@ -18,6 +18,7 @@ def get_model(cfg): d_model=cfg.dim, n_heads=cfg.heads, n_layers=cfg.layers, + n_experts=cfg.experts, config = cfg, )