mrq 83eab4fa59 actually going for the suggested "2x layers, no intermediate scaling" is wrong for VALL-E, directly copying the normal transformer structure fixes mamba2 performance in the test trainer		2024-06-13 20:08:22 -05:00
..
emb	ugh	2024-06-09 11:39:43 -05:00
engines	validated that inferencing works, changed some defaults (NAR benefits from greedy sampling)	2024-06-09 17:11:38 -05:00
ext	fixes	2024-06-04 00:07:00 -05:00
models	actually going for the suggested "2x layers, no intermediate scaling" is wrong for VALL-E, directly copying the normal transformer structure fixes mamba2 performance in the test trainer	2024-06-13 20:08:22 -05:00
utils	sanity cleanup, backup config yaml for each log file	2024-06-09 11:22:52 -05:00
__init__.py	Rewrite init	2023-08-02 21:53:35 +00:00
__main__.py	the NAR only dream is dead (it just won't work)	2024-06-12 19:49:47 -05:00
config.py	option to split classifier per-level instead of sharing one (at this point I'm just scrambling to try and cope with training a DAC model, the NAR is being a pain)	2024-06-11 22:28:59 -05:00
data.py	ugh	2024-06-11 23:59:28 -05:00
export.py	ugh	2024-06-09 11:39:43 -05:00
inference.py	the NAR only dream is dead (it just won't work)	2024-06-12 19:49:47 -05:00
plot.py	ugh	2024-06-09 11:39:43 -05:00
samplers.py	separated samplers into its own file, don't bother copying the logits back to the GPU after sampling, it's not necessary	2023-10-11 12:25:31 -05:00
train.py	ugh	2024-06-09 11:39:43 -05:00
webui.py	validated that inferencing works, changed some defaults (NAR benefits from greedy sampling)	2024-06-09 17:11:38 -05:00