def __add_bert_args()

in lama/options.py [0:0]

• 23 lines of code
• 1 McCabe index (conditional complexity)

def __add_bert_args(parser):
    group = parser.add_argument_group("BERT")
    group.add_argument(
        "--bert-model-dir",
        "--bmd",
        dest="bert_model_dir",
        help="directory that contains the BERT pre-trained model and the vocabulary",
    )
    group.add_argument(
        "--bert-model-name",
        "--bmn",
        dest="bert_model_name",
        default="bert-base-cased",
        help="name of the BERT pre-trained model (default = 'bert-base-cased')",
    )
    group.add_argument(
        "--bert-vocab-name",
        "--bvn",
        dest="bert_vocab_name",
        default="vocab.txt",
        help="name of vocabulary used to pre-train the BERT model (default = 'vocab.txt')",
    )
    return group