Update README.md

kaistAI · Oct 5, 2024 · 3212bae · 3212bae
1 parent 840bd7b
commit 3212bae
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -43,7 +43,7 @@ RUN_NAME=EXP_NAME
 
 export SCRATCH_DIR='PATH_TO_THE_REPOSITORY'
 
-python -m torch.distributed.run --nproc_per_node=4 ${SCRATCH_DIR}/scripts/train.py configs/official/OLMo-1B-105.yaml \
+python -m torch.distributed.run --nproc_per_node=4 ${SCRATCH_DIR}/scripts/train.py configs/official/OLMo-7B.yaml \
     --run_name=${RUN_NAME} \
     --data.paths=[${SCRATCH_DIR}/dolma_extracted/360000-363100.npy] \
     --load_path=PATH_TO_THE_CHECKPOINT/step${BASE_STEP}-unsharded \