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"