diff --git a/scripts/train/benchmarking/run_local.sh b/scripts/train/benchmarking/run_local.sh
index 3ca57f0840..1a881f10b6 100755
--- a/scripts/train/benchmarking/run_local.sh
+++ b/scripts/train/benchmarking/run_local.sh
@@ -7,4 +7,4 @@ IMAGE="mosaicml/pytorch:2.0.1_cu118-python3.10-ubuntu20.04"
 # A100 40GB
 CLUSTER_40GB=r1z1
 # seqlen 2048
-python submit_benchmarks.py --project $PROJECT -m 125m.yaml -g 8 --microbatch_size  24 --accum  3 --image $IMAGE --git_commit $GIT_COMMIT --gpu_type a100_40gb --cluster $CLUSTER_40GB -s 11 11 --RUN --LOCAL
\ No newline at end of file
+python submit_benchmarks.py --project $PROJECT -m 125m.yaml -g 8 --microbatch_size  24 --accum  3 --image $IMAGE --git_commit $GIT_COMMIT --gpu_type a100_80gb --cluster $CLUSTER_40GB -s 11 11 --RUN --LOCAL
\ No newline at end of file