33 lines
713 B
Bash
33 lines
713 B
Bash
|
#!/bin/bash
|
||
|
|
SUCCESS=0
|
||
|
|
|
||
|
|
IFS=$'\n' read -d '' -r -a MODEL_CONFIGS < "weight_loading/models.txt"
|
||
|
|
|
||
|
|
for MODEL_CONFIG in "${MODEL_CONFIGS[@]}"
|
||
|
|
do
|
||
|
|
LOCAL_SUCCESS=0
|
||
|
|
IFS=', ' read -r -a array <<< "$MODEL_CONFIG"
|
||
|
|
|
||
|
|
echo "=== RUNNING MODEL: $MODEL_CONFIG ==="
|
||
|
|
|
||
|
|
export QUANTIZATION=${array[0]}
|
||
|
|
export MODEL_NAME=${array[1]}
|
||
|
|
export REVISION=${array[2]}
|
||
|
|
pytest -s weight_loading/test_weight_loading.py || LOCAL_SUCCESS=$?
|
||
|
|
|
||
|
|
if [[ $LOCAL_SUCCESS == 0 ]]; then
|
||
|
|
echo "=== PASSED MODEL: ${MODEL_CONFIG} ==="
|
||
|
|
else
|
||
|
|
echo "=== FAILED MODEL: ${MODEL_CONFIG} ==="
|
||
|
|
fi
|
||
|
|
|
||
|
|
SUCCESS=$((SUCCESS + LOCAL_SUCCESS))
|
||
|
|
|
||
|
|
done
|
||
|
|
|
||
|
|
if [ "${SUCCESS}" -eq "0" ]; then
|
||
|
|
exit 0
|
||
|
|
else
|
||
|
|
exit 1
|
||
|
|
fi
|