From 7d711cd80dff17df78f8340df1b7616af0809407 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Mon, 23 Sep 2024 09:48:40 +0200 Subject: [PATCH] Update model_utils.py for new API (#199) --- src/alignment/model_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/alignment/model_utils.py b/src/alignment/model_utils.py index 6795b6a..650517e 100644 --- a/src/alignment/model_utils.py +++ b/src/alignment/model_utils.py @@ -22,7 +22,7 @@ from transformers.trainer_utils import get_last_checkpoint from accelerate import Accelerator from huggingface_hub import list_repo_files -from huggingface_hub.utils._errors import RepositoryNotFoundError +from huggingface_hub.errors import RepositoryNotFoundError from huggingface_hub.utils._validators import HFValidationError from peft import LoraConfig, PeftConfig