mrq d343bde09b residual_in_fp32=False for mamba arch backends because it breaks the classifier (output projection / lm head / what-have-you) under AMP		2024-06-15 12:08:03 -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	residual_in_fp32=False for mamba arch backends because it breaks the classifier (output projection / lm head / what-have-you) under AMP	2024-06-15 12:08:03 -05:00
utils	sampler update (some brainworm just never actually had a sampler for sample_type=path)	2024-06-14 16:55:40 -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	added option to sort paths by durations to better group equally lengthed sequences together (and there was maybe a logic error from creating the samplers and then interleave-reordering paths, desyncing them, maybe)	2024-06-13 22:37:34 -05:00
data.py	residual_in_fp32=False for mamba arch backends because it breaks the classifier (output projection / lm head / what-have-you) under AMP	2024-06-15 12:08:03 -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