File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed
benchmarks/float8/training Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -53,7 +53,7 @@ cd ${TORCHTITAN_ROOT}
5353echo " float8 args: ${FLOAT8_ARGS} "
5454
5555# run the command with the specified arguments
56- CONFIG_FILE=" ./torchtitan/models/llama3/train_configs/llama3_8b.toml" ${TORCHTITAN_ROOT} /run_train.sh --training.steps=${STEPS} --training.local-batch-size=${LOCAL_BATCH_SIZE} --training. compile ${FLOAT8_ARGS} ${EXTRA_ARGS} 2>&1 | tee ${LOG_FILE}
56+ CONFIG_FILE=" ./torchtitan/models/llama3/train_configs/llama3_8b.toml" ${TORCHTITAN_ROOT} /run_train.sh --training.steps=${STEPS} --training.local-batch-size=${LOCAL_BATCH_SIZE} --compile.enable ${FLOAT8_ARGS} ${EXTRA_ARGS} 2>&1 | tee ${LOG_FILE}
5757
5858# return to original working directory
5959cd $original_dir
You can’t perform that action at this time.
0 commit comments