Update model_utils.py for new API (#199)

huggingface · Sep 23, 2024 · 7d711cd · 7d711cd
1 parent 27f7dbf
commit 7d711cd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/alignment/model_utils.py b/src/alignment/model_utils.py
@@ -22,7 +22,7 @@
 
 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