mrq 2deb995cc9 updated setup script		2023-10-06 20:08:28 -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	fixed training stats not loading from exported weights, a bit of a readme cleanup, updated example training yaml	2023-09-23 19:59:00 -05:00
models	updated setup script	2023-10-06 20:08:28 -05:00
utils	implicitly load checkpoint if deepspeed checkpoint not found, updated setup script to grab the diskcached dataloader things	2023-10-06 10:02:45 -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	added compat flags for torchscale because the maintainer for torchscale broke compat for existing models	2023-10-05 16:39:46 -05:00
data.py	oops	2023-10-06 09:26:52 -05:00
export.py	ops	2023-08-20 13:42:18 -05:00
inference.py	restructured some things with the model to remove dead weights	2023-09-20 19:10:59 -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	revert the frankensteined "train one model but hotload the other" since it kept loading the last exported weights and I'm not supporting this usecase anymore anyways	2023-09-22 13:04:17 -05:00
webui.py	set default NAR levels to max for the web UI	2023-09-29 19:14:16 -05:00