from transformers import PretrainedConfig

class SwinCXRConfig(PretrainedConfig):
    model_type = "swin_cxr"

    def __init__(self, num_classes=3, embed_dim=128, num_heads=4, num_layers=4, dropout=0.1, **kwargs):
        self.num_classes = num_classes
        self.embed_dim = embed_dim
        self.num_heads = num_heads
        self.num_layers = num_layers
        self.dropout = dropout
        super().__init__(**kwargs)