File size: 1,028 Bytes
b3ccc39 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
# Source: https://huggingface.co/microsoft/BiomedVLP-CXR-BERT-specialized/blob/refs%2Fpr%2F5/configuration_cxrbert.py
# Date: 2024-07-01
# ------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License (MIT). See LICENSE in the repo root for license information.
# ------------------------------------------------------------------------------------------
from typing import Any
from transformers import BertConfig, BertTokenizer
class CXRBertConfig(BertConfig):
"""
Config class for CXR-BERT model.
:param projection_size: Dimensionality of the joint latent space.
"""
model_type = "cxr-bert"
def __init__(self, projection_size: int = 128, **kwargs: Any) -> None:
super().__init__(**kwargs)
self.projection_size = projection_size
class CXRBertTokenizer(BertTokenizer):
def __init__(self, **kwargs: Any) -> None:
super().__init__(**kwargs)
|