for para in $*
do
    if [[ $para == --profiling* ]];then
        profiling=${para#*=}
    fi
done

# Runs Mixtral 8x7B model
source /opt/dtk/env.sh
HOST=localhost
PORT=25900
DATA_PATH="path to my-mixtral_text_document"
TOKENIZER_MODEL_PATH="path to tokenizer.model"
CHECKPOINT_PATH="path to ckpt" 


mpirun -np 8  --allow-run-as-root \
              train_mixtral_8x7B_1nodes.sh \
              ${HOST} \
              ${PORT} \
              --data_path=$DATA_PATH \
              --tokenizer_path=$TOKENIZER_MODEL_PATH \
              --checkpoint_path=$CHECKPOINT_PATH \
              --profiling=$profiling > log-1nodes-`date +%F-%H%M`.log 2>&1

wait