update accelerate ver for schedule_free optimizers

Former-commit-id: 2de74e79049ce8e50f605f649275b1dbfb899c8c
2024-09-09 22:51:08 +08:00
parent 800567cde7
commit 588ea95732
4 changed files with 4 additions and 3 deletions
--- a/src/llamafactory/data/template.py
+++ b/src/llamafactory/data/template.py
@@ -357,6 +357,7 @@ def get_template_and_fix_tokenizer(tokenizer: "PreTrainedTokenizer", data_args:
        require_version(
            "transformers>=4.45.0.dev0", "To fix: pip install git+https://github.com/huggingface/transformers.git"
        )
+        require_version("accelerate>=0.34.0", "To fix: pip install accelerate>=0.34.0")

    if data_args.template is None:
        template = TEMPLATES["empty"]  # placeholder