From afa3c863918df12b2fe91355ae53ed44250ecfef Mon Sep 17 00:00:00 2001 From: hiyouga Date: Fri, 29 Dec 2023 04:53:36 +0800 Subject: [PATCH] fix version Former-commit-id: 47da742fc971520a697ed5ce611f8d6de61ab22b --- src/llmtuner/model/loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/model/loader.py b/src/llmtuner/model/loader.py index b7ec42ff..11cb4f72 100644 --- a/src/llmtuner/model/loader.py +++ b/src/llmtuner/model/loader.py @@ -60,7 +60,7 @@ def load_model_and_tokenizer( model = None if is_trainable and model_args.use_unsloth: - require_version("unsloth==2023.12", "Follow the instructions at: https://github.com/unslothai/unsloth") + require_version("unsloth", "Follow the instructions at: https://github.com/unslothai/unsloth") from unsloth import FastLlamaModel, FastMistralModel # type: ignore unsloth_kwargs = { "model_name": model_args.model_name_or_path,