mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-04 04:32:50 +08:00
modify weight name
Former-commit-id: 919acc2b0be03ede08cc0018784e8874a920e300
This commit is contained in:
parent
61960189b2
commit
b29c4fb308
@ -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"
|
||||
|
Loading…
x
Reference in New Issue
Block a user