simplify code

2023-07-20 15:08:57 +08:00
parent d1d8e8bae1
commit 67a2773074
18 changed files with 52 additions and 136 deletions
--- a/src/train_bash.py
+++ b/src/train_bash.py
@@ -1,4 +1,4 @@
-from llmtuner import get_train_args, run_pt, run_sft, run_rm, run_ppo
+from llmtuner.tuner import get_train_args, run_pt, run_sft, run_rm, run_ppo


 def main():