Spaces:
No application file
No application file
export ALBERT_BASE_DIR=$1 | |
transformers-cli convert --model_type albert \ | |
--tf_checkpoint $ALBERT_BASE_DIR/tf_model \ | |
--config $ALBERT_BASE_DIR/config.json \ | |
--pytorch_dump_output $ALBERT_BASE_DIR/pytorch_model.bin | |