adds auto adapter merge to dpo script

This commit is contained in:
edbeeching
2023-11-10 14:15:44 +01:00
parent 54185783e0
commit 7f1a14e0d4
2 changed files with 33 additions and 4 deletions
+5 -1
View File
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, BitsAndBytesConfig, PreTrainedTokenizer
from accelerate import Accelerator
from peft import LoraConfig, PeftConfig
from huggingface_hub import list_repo_files
from .configs import DataArguments, ModelArguments
from .data import DEFAULT_CHAT_TEMPLATE
@@ -77,3 +77,7 @@ def get_peft_config(model_args: ModelArguments) -> PeftConfig | None:
)
return peft_config
def is_adapter_model(model_name_or_path: str, revision: str = "main") -> bool:
repo_files = list_repo_files(model_name_or_path, revision=revision)
return "adapter_model.safetensors" in repo_files or "adapter_model.bin" in repo_files