From ebb32e85f8f9cf1af43e8355d999975a55726dd2 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: dd7500b65d0d548441eece101b60d51fa619cc0f --- 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,