From ce43386080fa1535672f4d879ffe6c4360a1ef7d Mon Sep 17 00:00:00 2001 From: hiyouga Date: Wed, 7 Jun 2023 12:08:39 +0800 Subject: [PATCH] tiny fix --- src/utils/common.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/utils/common.py b/src/utils/common.py index 02880ed8..eaf1d005 100644 --- a/src/utils/common.py +++ b/src/utils/common.py @@ -101,7 +101,7 @@ def _init_adapter( logger.info("Fine-tuning method: LoRA") lastest_checkpoint = None - assert os.path.exists(model_args.checkpoint_dir[0], CONFIG_NAME), \ + assert os.path.exists(os.path.join(model_args.checkpoint_dir[0], CONFIG_NAME)), \ "The given checkpoint is not a LoRA checkpoint, please specify `--finetuning_type full/freeze` instead." if model_args.checkpoint_dir is not None: