From b29c4fb3081e5c18039d7f8d420831ed05638246 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Tue, 9 Jan 2024 20:22:47 +0800 Subject: [PATCH] modify weight name Former-commit-id: 919acc2b0be03ede08cc0018784e8874a920e300 --- src/llmtuner/extras/constants.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/llmtuner/extras/constants.py b/src/llmtuner/extras/constants.py index a81db3a7..e24712c4 100644 --- a/src/llmtuner/extras/constants.py +++ b/src/llmtuner/extras/constants.py @@ -40,9 +40,9 @@ TRAINING_STAGES = { "Pre-Training": "pt" } -V_HEAD_WEIGHTS_NAME = "v_head.bin" +V_HEAD_WEIGHTS_NAME = "value_head.bin" -V_HEAD_SAFE_WEIGHTS_NAME = "v_head.safetensors" +V_HEAD_SAFE_WEIGHTS_NAME = "value_head.safetensors" class DownloadSource(str, Enum): DEFAULT = "hf"