Skip to content

Commit d9b394a

Browse files
authored
Clean up Voxtral Realtime HF weights after export (pytorch#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.
1 parent b40d6fe commit d9b394a

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)