Commit Graph

21 Commits

Author SHA1 Message Date
Colin 7d16743184 enable pretrain. 2024-02-22 15:03:32 +08:00
周以晴 b655153ec7
Merge pull request #2 from Yiqing-Zhou/fix-custom-models
[fix] fix genarate with custom models does not go to custom_models
2023-05-28 22:58:51 +08:00
yiqing-zhou 9f8f9ecc89 [fix] fix genarate with custom models does not go to custom_models 2023-05-28 22:57:51 +08:00
周以晴 e8d543558c
Merge pull request #1 from Yiqing-Zhou/custom-model-configs
[feature] custom model configs
2023-05-28 21:48:24 +08:00
yiqing-zhou fcb93e52c4 [feature] custom model configs 2023-05-28 21:39:51 +08:00
yiqing-zhou b76d333f39 [code] formatter-caused changes 2023-05-28 20:02:56 +08:00
周以晴 10a88a5012
Update README.md 2023-05-14 23:08:44 +08:00
Yiqing-Zhou 30df20402d [code] update .vscode launch.json 2023-05-14 22:55:01 +08:00
Yiqing-Zhou 6827898339 . 2023-05-14 22:53:28 +08:00
Yiqing-Zhou 216bc4643c [feature] custom_models 2023-05-14 22:23:16 +08:00
Yiqing-Zhou 5e6b747baf [fix] add patch to fix DeepSpeedStrategy offload 'zero_force_ds_cpu_optimizer' issue 2023-05-09 23:00:28 +08:00
Yiqing-Zhou 8a5e2043bb [optimize] map_location='cpu' for load_from_checkpoint 2023-05-09 00:37:52 +08:00
Yiqing-Zhou 3f92bbbaa2 [feature] new args learning_rate max_epochs 2023-05-09 00:35:28 +08:00
周以晴 dc2941f790
Create LICENSE 2023-05-08 00:26:38 +08:00
Yiqing-Zhou 70ff2acaf0 [fix] add patch to fix FSDPStrategy checkpoint issue 2023-05-07 16:51:57 +08:00
Yiqing-Zhou 5392a845f7 [feature] export model checkpoint from pl.LightningModule 2023-05-07 13:18:58 +08:00
Yiqing-Zhou 09507449f7 [code] refactor 2023-05-07 13:01:02 +08:00
Yiqing-Zhou 939be31c10 [feature] new arg use_tril_attention_mask 2023-05-06 21:06:18 +08:00
Yiqing-Zhou 0324eb4103 [code] update requirements 2023-05-06 21:05:53 +08:00
Yiqing-Zhou 9b7f9b9d60 [feature] persistent_workers; new args accumulate_grad_batches strategy 2023-05-06 19:39:24 +08:00
Yiqing-Zhou 45fa065530 Initial Commit 2023-05-04 21:52:25 +08:00