mirror of
https://github.com/wassname/alignment-handbook.git
synced 2026-06-27 14:58:48 +08:00
Update model_utils.py for new API (#199)
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user