Commit Graph

9 Commits

Author SHA1 Message Date
Tom Hunn
a4ca14b5f6 add and run pre-commit with ruff 2024-10-21 14:46:45 +10:00
SWivid
28b46d32d3 rewrite without einx & einops; clean up 2024-10-20 20:41:24 +08:00
SWivid
554f3189e1 Use default fp16 inference 2024-10-20 16:32:18 +08:00
cocktailpeanut
ae6e97b836 user-friendly wandb support 2024-10-18 14:59:59 -04:00
Jarod Mica
31e5051d51 default to weights_only=True for safer loading 2024-10-15 00:37:46 -07:00
SWivid
46d391a876 fix replacement of ckpt keys when do finetune training 2024-10-13 17:20:18 +08:00
SWivid
0d7b47bc3b enable correct ckpt loading for finetune 2024-10-13 14:41:08 +08:00
Zhikang Niu
edc189fa96 Update trainer.py 2024-10-13 10:04:13 +08:00
SWivid
074881635d basic 2024-10-08 21:56:51 +08:00