5 Commits

Author	SHA1	Message	Date
oobabooga	98af4bfb0d	Refactor the code to make it more modular	2023-02-23 12:05:25 -03:00
oobabooga	e52b697d5a	Add bf16 back here (the fp16 -> bf16 conversion takes a few seconds)	2023-02-21 00:54:53 -03:00
oobabooga	c1de491c63	No need to have bf16 support here	2023-02-20 15:12:42 -03:00
oobabooga	8123c5b01a	Sort the imports	2023-02-20 15:06:27 -03:00
oobabooga	e195377050	Deprecate torch dumps, move to safetensors (they load even faster)	2023-02-20 15:03:19 -03:00