We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d6d7caf commit 482f8cfCopy full SHA for 482f8cf
.ci/scripts/validate.sh
@@ -110,13 +110,13 @@ function generate_compiled_model_output() {
110
function generate_aoti_model_output() {
111
local CHECKPOINT_PATH="$1"
112
local TARGET_DEVICE="${2:-cpu}"
113
- local DTYPES="${3}"
+ local DTYPES="${3:-default}"
114
local MODEL_DIR="${CHECKPOINT_PATH%/*}"
115
local MODEL_NAME=$(basename "$CHECKPOINT_PATH" | sed 's/\.[^.]*$//')
116
117
echo "Local DTYPES=$DTYPES"
118
119
- if [[ DTYPES == "default" ]]; then
+ if [[ $DTYPES == "default" ]]; then
120
if [[ $CHECKPOINT_PATH != *"stories"* && $TARGET_DEVICE == "cuda" ]]; then
121
DTYPES="bfloat16"
122
EXCLUDE_INT8_QUANT=true
0 commit comments