mrq ccb14c06ef mamba2-hf using vasqu/mamba2-torch because it lets me use mamba2 without triton ops (training with my 4xV100s are not happy with mamba2 because of triton)		2024-06-14 19:42:17 -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	mamba2-hf using vasqu/mamba2-torch because it lets me use mamba2 without triton ops (training with my 4xV100s are not happy with mamba2 because of triton)	2024-06-14 19:42:17 -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	sampler update (some brainworm just never actually had a sampler for sample_type=path)	2024-06-14 16:55:40 -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