8 Commits

Author	SHA1	Message	Date
oobabooga	ea6e77df72	Make the code more like PEP8 for readability (#862)	2023-04-07 00:15:45 -03:00
oobabooga	7224343a70	Improve the imports	2023-02-23 14:41:42 -03:00
oobabooga	1dacd34165	Further refactor	2023-02-23 13:28:30 -03:00
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