Skip to content
This repository was archived by the owner on Feb 3, 2025. It is now read-only.

Commit 7086a97

Browse files
Update base_run_inference.sh
1 parent 7a40ba9 commit 7086a97

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tftrt/benchmarking-python/huggingface/gpt2/base_run_inference.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ BASE_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
88
MODEL_NAME=""
99

1010
# Default Argument Values
11-
BATCH_SIZE=32
11+
BATCH_SIZE=12
1212
SEQ_LEN=1024
1313
VOCAB_SIZE=50257
1414

0 commit comments

Comments
 (0)