Method __init__

tensorrt_llm/models/bert/model.py:375–381

(self, config: BERTConfig)

class BertForQuestionAnswering(BertBase):

    def __init__(self, config: BERTConfig):
        super().__init__(config)
        self.bert = BertModel(config)
        self.num_labels = config.num_labels
        self.qa_outputs = Linear(config.hidden_size,
                                 config.num_labels,
                                 dtype=config.dtype)

    def forward(self,
                input_ids=None,