File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -204,7 +204,7 @@ jobs:
204204 if [ -f "$DOWNLOADED_TOKENIZER_FILE_PATH" ]; then
205205 echo "${TOKENIZER_FILE} downloaded successfully at: $DOWNLOADED_TOKENIZER_FILE_PATH"
206206 python -m extension.llm.tokenizer.tokenizer -t $DOWNLOADED_TOKENIZER_FILE_PATH -o ./${TOKENIZER_BIN_FILE}
207- ls ./tokenizer.bin
207+ ls "${TOKENIZER_BIN_FILE}"
208208 else
209209 echo "Failed to download ${TOKENIZER_FILE} from ${{ matrix.model }}."
210210 exit 1
@@ -213,7 +213,10 @@ jobs:
213213 MODEL_NAME=$(echo "${{ matrix.model }}" | sed 's,/,-,g')
214214 python -m extension.export_util.export_hf_model -hfm=${{ matrix.model }} -o "${MODEL_NAME}_xnnpack_fp32"
215215
216- cp tokenizer.bin *.pte "${ARTIFACTS_DIR_NAME}"
216+ # Prepare the model to upload
217+ zip -j model.zip *.pte tokenizer.bin
218+ mkdir -p "${ARTIFACTS_DIR_NAME}"
219+ mv model.zip "${ARTIFACTS_DIR_NAME}"
217220
218221 elif [[ ${{ matrix.model }} =~ ^stories* ]]; then
219222 # Install requirements for export_llama
You can’t perform that action at this time.
0 commit comments