support unsloth

Former-commit-id: b857f00234b90b785d82ca7cdb29af3d948b1a7b
2023-12-23 00:14:33 +08:00
parent 1066898e32
commit 6faf9c35a9
11 changed files with 224 additions and 171 deletions
--- a/src/llmtuner/model/parser.py
+++ b/src/llmtuner/model/parser.py
@@ -104,8 +104,7 @@ def get_train_args(args: Optional[Dict[str, Any]] = None) -> _TRAIN_CLS:

    # Setup logging
    if training_args.should_log:
-        log_level = training_args.get_process_log_level()
-        _set_transformers_logging(log_level)
+        _set_transformers_logging()

    # Check arguments
    data_args.init_for_training(training_args.seed)