mirror of
https://github.com/wassname/alignment-handbook.git
synced 2026-07-20 12:10:44 +08:00
adds updated model paths, adds eval to sft scripts
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
# Model arguments
|
||||
model_name_or_path: lewtun/zephyr-7b-sft
|
||||
model_name_or_path: alignment-handbook/zephyr-7b-sft-full
|
||||
|
||||
# Data training arguments
|
||||
# For definitions, see: src/h4/training/config.py
|
||||
|
||||
@@ -1,6 +1,5 @@
|
||||
# Model arguments
|
||||
model_name_or_path: HuggingFaceH4/mistral-7b-ift
|
||||
model_revision: v14.0
|
||||
model_name_or_path: alignment-handbook/zephyr-7b-sft-lora
|
||||
torch_dtype: auto
|
||||
|
||||
# LORA
|
||||
|
||||
@@ -14,6 +14,7 @@ preprocessing_num_workers: 12
|
||||
|
||||
# SFT trainer config
|
||||
bf16: true
|
||||
do_eval: true
|
||||
evaluation_strategy: epoch
|
||||
gradient_accumulation_steps: 2
|
||||
gradient_checkpointing: true
|
||||
|
||||
@@ -24,6 +24,7 @@ preprocessing_num_workers: 12
|
||||
|
||||
# SFT trainer config
|
||||
bf16: true
|
||||
do_eval: true
|
||||
evaluation_strategy: epoch
|
||||
gradient_accumulation_steps: 128
|
||||
ddp_find_unused_parameters: true
|
||||
|
||||
Reference in New Issue
Block a user