File size: 401 Bytes
b2ebc95 | 1 2 3 4 5 6 7 8 9 10 11 | #!/bin/bash
# finalize_ckpt.sh <ckpt_dir> — add the files prime-rl's trainer does not save but that
# prime-rl's own inference server (and any processor-loading path) needs.
set -uo pipefail
B=/var/lib/agentptb-cache/a/models/base
D="${1:?checkpoint dir}"
for f in preprocessor_config.json video_preprocessor_config.json merges.txt vocab.json; do
[ -f "$D/$f" ] || cp "$B/$f" "$D/$f"
done
ls "$D"
|