mrq ec25f56bd9 used torch.max fixes things, somehow, for dynamic temp sampling		2023-10-10 16:42:24 -05:00
..
emb	(need to verify) added modifying model size and config bool to align with VALL-E continuous' methodology	2023-09-01 17:19:34 -05:00
engines	cleanup, use deepspeed inferencing pathway if requested	2023-10-09 15:24:04 -05:00
models	used torch.max fixes things, somehow, for dynamic temp sampling	2023-10-10 16:42:24 -05:00
utils	cleanup, use deepspeed inferencing pathway if requested	2023-10-09 15:24:04 -05:00
__init__.py	Rewrite init	2023-08-02 21:53:35 +00:00
__main__.py	added mirostat sampling (given a partially trained model, it got far decent output than I expected, need to test on a better trained model)	2023-09-18 18:55:41 -05:00
config.py	cleanup, use deepspeed inferencing pathway if requested	2023-10-09 15:24:04 -05:00
data.py	trim the input prompt to 3 seconds when training NAR tasks (marked as experimental; the paper mentions doing so, but I don't know how much this would harm the retention heads)	2023-10-09 22:03:58 -05:00
export.py	cleanup, use deepspeed inferencing pathway if requested	2023-10-09 15:24:04 -05:00
inference.py	cleanup, use deepspeed inferencing pathway if requested	2023-10-09 15:24:04 -05:00
plot.py	added min-x and min-y arguments to plot.py, helper script to download from my existing checkpoint	2023-10-04 19:41:37 -05:00
train.py	cleanup, use deepspeed inferencing pathway if requested	2023-10-09 15:24:04 -05:00
webui.py	reduced dynamic temperature threshold to > 1.0, as it seems to not quite be useful for audio LMs, sped up any sampling that touches logits by copying them to CPU first, as accessing tensors on the GPU is slow as balls)	2023-10-09 14:46:17 -05:00