wassname/alignment-handbook
1
0
Fork 0
mirror of https://github.com/wassname/alignment-handbook.git synced 2026-06-27 17:47:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update model_utils.py for new API (#199)

Browse Source
This commit is contained in:
Kashif Rasul
2024-09-23 09:48:40 +02:00
committed by GitHub
parent 27f7dbf006
commit 7d711cd80d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/alignment/model_utils.py
+1 -1
View File
@@ -22,7 +22,7 @@ from transformers.trainer_utils import get_last_checkpoint
from accelerate import Accelerator from accelerate import Accelerator
from huggingface_hub import list_repo_files 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 huggingface_hub.utils._validators import HFValidationError
from peft import LoraConfig, PeftConfig from peft import LoraConfig, PeftConfig