mrq 1d159b1476 updated export routine to split LoRA weights from the state dict (should work with deepspeed)		2024-06-17 13:28:18 -05:00
..
emb	ugh	2024-06-09 11:39:43 -05:00
engines	updated export routine to split LoRA weights from the state dict (should work with deepspeed)	2024-06-17 13:28:18 -05:00
ext	fixes	2024-06-04 00:07:00 -05:00
models	updated export routine to split LoRA weights from the state dict (should work with deepspeed)	2024-06-17 13:28:18 -05:00
utils	naive, rudimentary DeepSpeed support (just live with the LoRA weights living with the original weights, they can be split later)	2024-06-17 13:17:24 -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	updated export routine to split LoRA weights from the state dict (should work with deepspeed)	2024-06-17 13:28:18 -05:00
data.py	ugh	2024-06-15 12:29: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