mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

experts weren't forwarded into constructer (wasted a few days of training garbage)

Browse Source
This commit is contained in:
mrq 2023-12-23 16:08:17 -06:00
parent 0db3203b21
commit e513d2ef19
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
vall_e/models/__init__.py
View File

@ -18,6 +18,7 @@ def get_model(cfg):
d_model=cfg.dim, d_model=cfg.dim,
n_heads=cfg.heads, n_heads=cfg.heads,
n_layers=cfg.layers, n_layers=cfg.layers,
n_experts=cfg.experts,
config = cfg, config = cfg,
) )