adds updated model paths, adds eval to sft scripts

This commit is contained in:
edbeeching
2023-11-10 09:26:39 +01:00
parent 4b0769d137
commit 13141a4b0b
4 changed files with 4 additions and 3 deletions
+1 -1
View File
@@ -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 -2
View File
@@ -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
+1
View File
@@ -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
+1
View File
@@ -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