Skip to content

Commit 0186941

Browse files
committed
Clean up Voxtral Realtime HF weights after export
The downloaded model_weights/ directory (~20GB) was not removed after export, causing it to be uploaded to S3 as part of the CI artifact. Delete it after copying out the tokenizer since the weights are fully baked into the .pte at that point.
1 parent 5e265e6 commit 0186941

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

.ci/scripts/export_model_artifact.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -373,6 +373,7 @@ if [ "$MODEL_NAME" = "voxtral_realtime" ]; then
373373
fi
374374
# Copy tokenizer from downloaded model weights
375375
cp "$LOCAL_MODEL_DIR/tekken.json" "${OUTPUT_DIR}/tekken.json"
376+
rm -rf "$LOCAL_MODEL_DIR"
376377
ls -al "${OUTPUT_DIR}"
377378
echo "::endgroup::"
378379
exit 0

0 commit comments

Comments
 (0)