Method __init__

tensorrt_llm/models/bert/model.py:127–128

(self, config: BERTConfig)

    config_class = BERTConfig

    def __init__(self, config: BERTConfig):
        super().__init__(config)

    @classmethod
    def load_hf_bert(cls, model_dir: str, load_model_on_cpu: bool,