Former-commit-id: ce43386080
This commit is contained in:
hiyouga
2023-06-07 12:08:39 +08:00
parent ce08b4a7ec
commit 93f2e35035

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: