Skip to content

Commit 4674a34

Browse files
Clean up Voxtral Realtime HF weights after export (#18357)
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. Co-authored-by: Mergen Nachin <mnachin@meta.com>
1 parent 2bba833 commit 4674a34

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
@@ -347,6 +347,7 @@ if [ "$MODEL_NAME" = "voxtral_realtime" ]; then
347347
fi
348348
# Copy tokenizer from downloaded model weights
349349
cp "$LOCAL_MODEL_DIR/tekken.json" "${OUTPUT_DIR}/tekken.json"
350+
rm -rf "$LOCAL_MODEL_DIR"
350351
ls -al "${OUTPUT_DIR}"
351352
echo "::endgroup::"
352353
exit 0

0 commit comments

Comments
 (0)