Skip to content

Commit d4e417a

Browse files
mergennachinpytorchbot
authored andcommitted
Clean up Voxtral Realtime HF weights after export (#18289)
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. (cherry picked from commit d9b394a)
1 parent 2bba833 commit d4e417a

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)