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: