From 612ba26c4ccc80da260eab05340dd650834ff431 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Wed, 24 Apr 2024 05:10:07 +0800 Subject: [PATCH] fix bug Former-commit-id: 8f44dce08aa809bb7d4ea0bd5f48ca1c56436044 --- src/llmtuner/model/utils/unsloth.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/model/utils/unsloth.py b/src/llmtuner/model/utils/unsloth.py index 6c5f506f..974b41c0 100644 --- a/src/llmtuner/model/utils/unsloth.py +++ b/src/llmtuner/model/utils/unsloth.py @@ -73,7 +73,7 @@ def load_unsloth_peft_model( """ from unsloth import FastLanguageModel - unsloth_kwargs = _get_unsloth_kwargs(config, model_args.adapter_name_or_path, model_args) + unsloth_kwargs = _get_unsloth_kwargs(config, model_args.adapter_name_or_path[0], model_args) try: model, _ = FastLanguageModel.from_pretrained(**unsloth_kwargs) except NotImplementedError: