Former-commit-id: 7115bee4310888ec2e5f104e8d2c1f7127fb6ce6
This commit is contained in:
hiyouga 2023-06-07 12:08:39 +08:00
parent b9feb82e4e
commit 5e2ec2d104

View File

@ -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: