mrq 9910c75d5a checkpointing for bitnet impl		2024-05-12 07:52:54 -05:00
..
emb	ughh	2024-05-12 07:30:59 -05:00
engines	nan loss detection (should have added it earlier), loss scaling for local backend + fp16	2024-05-11 22:23:29 -05:00
ext
models	checkpointing for bitnet impl	2024-05-12 07:52:54 -05:00
utils	leverage between xformers and torch.backends.cuda.sdp_kernel for attention	2024-05-11 17:14:05 -05:00
__init__.py
__main__.py
config.py	ugh	2024-05-11 22:58:38 -05:00
data.py	ughh	2024-05-12 07:30:59 -05:00
export.py
inference.py
plot.py
samplers.py
train.py
webui.py