cd ../../

if [ -f "./scripts/env/env.sh" ]; then
    source ./scripts/env/env.sh
fi

# python /work/jf381/code/lm-research/main.py dataset=tinygsm_no_slides dataset.component_whitelist=null "$@"

# python /work/jf381/code/lm-research/main.py dataset=tinygsm_no_slides_bert  dataset.component_whitelist=null "$@"

python /work/jf381/code/lm-research/main.py dataset=slimpajama_60b  dataset.component_whitelist=null "$@"

# bert_pretrain
# python /work/jf381/code/lm-research/main.py dataset=slimpajama_5m  dataset.component_whitelist=null "$@"

# python /work/jf381/code/lm-research/main.py dataset=slimpajama_5m  dataset.component_whitelist=null "$@"