From 6c59994cdac01c8dc81f4e7ad19ade4d442f14a5 Mon Sep 17 00:00:00 2001 From: Will Constable Date: Fri, 12 Jul 2024 17:14:12 -0700 Subject: [PATCH] Update [ghstack-poisoned] --- create_seed_checkpoint.sh | 1 - run_llama_train.sh | 1 - run_memory_estimation.sh | 1 - 3 files changed, 3 deletions(-) diff --git a/create_seed_checkpoint.sh b/create_seed_checkpoint.sh index 3dfbde714..77185bfce 100755 --- a/create_seed_checkpoint.sh +++ b/create_seed_checkpoint.sh @@ -18,7 +18,6 @@ set -ex -TRAINER_DIR=${1:-/home/$USER/local/torchtitan} NGPU=1 LOG_RANK=0 CONFIG_FILE=${CONFIG_FILE:-"./train_configs/debug_model.toml"} diff --git a/run_llama_train.sh b/run_llama_train.sh index 56a12c03c..1114bae37 100755 --- a/run_llama_train.sh +++ b/run_llama_train.sh @@ -12,7 +12,6 @@ set -ex # LOG_RANK=0,1 NGPU=4 ./run_llama_train.sh NGPU=${NGPU:-"8"} NNODES=${NNODES:-"1"} -TRAINER_DIR=${TRAINER_DIR:-/home/$USER/local/torchtitan} LOG_RANK=${LOG_RANK:-0} CONFIG_FILE=${CONFIG_FILE:-"./train_configs/debug_model.toml"} diff --git a/run_memory_estimation.sh b/run_memory_estimation.sh index 96a620f41..c4d703ff2 100755 --- a/run_memory_estimation.sh +++ b/run_memory_estimation.sh @@ -12,7 +12,6 @@ set -ex # LOG_RANK=0,1 NGPU=4 ./run_llama_train.sh NGPU=${NGPU:-"8"} NNODES=${NNODES:-"1"} -TRAINER_DIR=${TRAINER_DIR:-/home/$USER/local/torchtitan} LOG_RANK=${LOG_RANK:-0} CONFIG_FILE=${CONFIG_FILE:-"./train_configs/debug_model.toml"}