mrq b0158a61d5 fixed some logic errors with training (grabbing wrong quant level...)		2024-06-07 20:34:36 -05:00
..
emb	DAC just doesn't work well enough......	2024-05-25 11:07:52 -05:00
engines	ugh	2024-06-06 21:57:11 -05:00
ext	fixes	2024-06-04 00:07:00 -05:00
models	fixed some logic errors with training (grabbing wrong quant level...)	2024-06-07 20:34:36 -05:00
utils	ugh	2024-06-06 21:57:11 -05:00
__init__.py	Rewrite init	2023-08-02 21:53:35 +00:00
__main__.py	re-added loading multiple models because I'm now entertaining having split AR/NAR models again (and need a way to load both at once)	2024-06-06 09:48:43 -05:00
config.py	I forgot the actual reason I was cleaning things up was to re-include prom loss calculation (I realized the reason I did this was because of an prom embedding oversight, it seems to work now)	2024-06-07 20:29:25 -05:00
data.py	ugh	2024-06-06 21:57:11 -05:00
export.py	added a flag to convert to a HF compatible model on export by stitching things	2024-06-03 22:34:47 -05:00
inference.py	finally got around to removing omegaconf	2024-06-07 20:23:53 -05:00
plot.py	deprecate sole AR/NAR model by only keeping the AR+NAR (the beauty of no one using this is that I can break compat as much as I want), add tone token for when I classify my dataset with tone/emotion in the future, some other things	2024-04-15 19:54:32 -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-06 21:57:11 -05:00
webui.py	re-added loading multiple models because I'm now entertaining having split AR/NAR models again (and need a way to load both at once)	2024-06-06 09:48:43 -05:00