mrq 589cfb0e18 yuge speedup because of a dumb oversight		2025-03-20 17:39:41 -05:00
..
emb
engines
models	yuge speedup because of a dumb oversight	2025-03-20 17:39:41 -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