def METHOD_NAME(parser):
    """Add model-specific arguments to the parser."""
    # fmt: off
    parser.add_argument('--embed-dim', type=int, metavar='N',
                        help='embedding dimension')
    parser.add_argument('--num-attention-heads', type=int, metavar='N',
                        help='num attention heads')
    parser.add_argument('--num-layers', type=int, metavar='N',
                        help='num layers')
    parser.add_argument('--dropout', type=float, metavar='D',
                        help='dropout probability for all fully connected layers '
                             'in the embeddings, encoder, and pooler')
    parser.add_argument('--attention-dropout', type=float, metavar='D',
                        help='dropout probability for attention weights')
    # fmt: on