mirror of
https://github.com/openai/gpt-2
synced 2025-08-22 02:02:01 +00:00
14 lines
410 B
Bash
14 lines
410 B
Bash
if [ "$#" -ne 1 ]; then
|
|
echo "You must enter the model name as a parameter, e.g.: sh download_model.sh 117M"
|
|
exit 1
|
|
fi
|
|
|
|
model=$1
|
|
|
|
mkdir -p models/$model
|
|
|
|
# TODO: gsutil rsync -r gs://gpt-2/models/ models/
|
|
for filename in checkpoint encoder.json hparams.json model.ckpt.data-00000-of-00001 model.ckpt.index model.ckpt.meta vocab.bpe; do
|
|
gsutil cp gs://gpt-2/models/$model/$filename models/$model
|
|
done
|