mrq 8068f24e35 cleaned up parallel nar, i think it's slightly faster but even the smallest model is still slower than ar+nar-len-llama-8...		2025-03-20 15:56:15 -05:00
..
emb
engines
models	cleaned up parallel nar, i think it's slightly faster but even the smallest model is still slower than ar+nar-len-llama-8...	2025-03-20 15:56:15 -05:00
utils
__init__.py
__main__.py
config.py	more tweaks to the new implementation (properly trim the len stuff to save some params, decoder to d_ffn expansion to 2 to maybe also make it faster, etc.)	2025-03-18 19:34:37 -05:00
data.py	another dataloader optimization	2025-03-15 20:18:58 -05:00
demo.py
export.py
inference.py
metrics.py
plot.py
samplers.py
train.py
webui.py	more tweaks to the new implementation (properly trim the len stuff to save some params, decoder to d_ffn expansion to 2 to maybe also make it faster, etc.)	2025-03-18 19:34:37 -05:00