Skip to content

Commit 482f8cf

Browse files
committed
bug fix
1 parent d6d7caf commit 482f8cf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.ci/scripts/validate.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,13 +110,13 @@ function generate_compiled_model_output() {
110110
function generate_aoti_model_output() {
111111
local CHECKPOINT_PATH="$1"
112112
local TARGET_DEVICE="${2:-cpu}"
113-
local DTYPES="${3}"
113+
local DTYPES="${3:-default}"
114114
local MODEL_DIR="${CHECKPOINT_PATH%/*}"
115115
local MODEL_NAME=$(basename "$CHECKPOINT_PATH" | sed 's/\.[^.]*$//')
116116

117117
echo "Local DTYPES=$DTYPES"
118118

119-
if [[ DTYPES == "default" ]]; then
119+
if [[ $DTYPES == "default" ]]; then
120120
if [[ $CHECKPOINT_PATH != *"stories"* && $TARGET_DEVICE == "cuda" ]]; then
121121
DTYPES="bfloat16"
122122
EXCLUDE_INT8_QUANT=true

0 commit comments

Comments
 (0)