mrq 94cf81d38c tweak		2024-09-05 23:21:18 -05:00
..
emb	moved prints to use logger, edited readme (fused_attn doesnt seem stable for training)	2024-08-29 13:27:16 -05:00
engines	fixes	2024-09-05 21:42:59 -05:00
ext	fixes	2024-06-04 00:07:00 -05:00
models	tweak	2024-09-05 23:21:18 -05:00
utils	tweak	2024-09-05 23:21:18 -05:00
__init__.py	Rewrite init	2023-08-02 21:53:35 +00:00
__main__.py	tweak	2024-09-05 23:21:18 -05:00
config.py	experimental implementation of STT (need to actually test on a model, test trainer seems to work)	2024-09-05 20:43:20 -05:00
data.py	tweak	2024-09-05 23:21:18 -05:00
demo.py	moved prints to use logger, edited readme (fused_attn doesnt seem stable for training)	2024-08-29 13:27:16 -05:00
export.py	added export option to convert Llama to MixtralMoE for another dumb experiment	2024-08-04 20:25:06 -05:00
inference.py	tweak	2024-09-05 23:21:18 -05:00
plot.py	ugh	2024-06-09 11:39:43 -05:00
samplers.py	possible speedup for samplers that require a list of previous tokens (the DRY sampler made me realize that I should copy the tolist() thing from the rep pen sampler for everything else)	2024-07-29 20:23:26 -05:00
train.py	moved prints to use logger, edited readme (fused_attn doesnt seem stable for training)	2024-08-29 13:27:16 -05:00
webui.py	tweak	2024-09-05 23:21:18 -05:00