mrq 692d09f9c1 eval/validation fix for SpeechX tasks		2024-07-19 09:16:37 -05:00
..
emb	fixes...	2024-07-18 23:25:32 -05:00
engines	mechanism to store the model config inside the weights and load them, some other things to allow LoRA training on the RetNet (gradient checkpointing will gripe about inputs not having require_grad and nothing seems to remedy it)	2024-07-16 18:23:13 -05:00
ext
models	fixes...	2024-07-18 23:25:32 -05:00
utils	oops, kept forgetting to actually pass in lang/tone tokens (despite not really using these at the moment)	2024-07-18 14:18:34 -05:00
__init__.py
__main__.py
config.py	fixes for re-introducing SpeechX tasks (need to actually validate if these all do the right things)	2024-07-18 17:16:32 -05:00
data.py	fixes...	2024-07-18 23:25:32 -05:00
export.py	mechanism to store the model config inside the weights and load them, some other things to allow LoRA training on the RetNet (gradient checkpointing will gripe about inputs not having require_grad and nothing seems to remedy it)	2024-07-16 18:23:13 -05:00
inference.py	allow loading a different model within the web ui (apparently I did not have the web UI in the documentation)	2024-07-15 19:59:48 -05:00
plot.py
samplers.py
train.py	eval/validation fix for SpeechX tasks	2024-07-19 09:16:37 -05:00
webui.py	allow loading a different model within the web ui (apparently I did not have the web UI in the documentation)	2024-07-15 19:59:48 -05:00