42e commited on
Commit
2697a62
·
verified ·
1 Parent(s): 1c0986b

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. configs/verifier_v_7b_fast.yaml +14 -0
configs/verifier_v_7b_fast.yaml ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # FAST 7B verifier training (~30-45 min) via sequence packing + FlashAttention-2.
2
+ # Packing concatenates the short rows (p50=602) into dense 2560-token sequences,
3
+ # cutting ~8127 padded rows to ~2000 dense ones -> far fewer steps. Packing REQUIRES
4
+ # FlashAttention-2 (else it attends across packed examples and corrupts training):
5
+ # pip install flash-attn --no-build-isolation # ~5-15 min build on Colab
6
+ # python -m mathcompose.train.train --task v --config configs/verifier_v_7b_fast.yaml \
7
+ # --gpu-preset none --push --hub-model-id <you>/mathcompose-verifier-7b
8
+ extends: "verifier_v_7b.yaml"
9
+
10
+ model:
11
+ attn_implementation: "flash_attention_2"
12
+
13
+ train:
14
+ packing: true