Upload folder using huggingface_hub
Browse files
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
|