from transformers import PretrainedConfig


class Swin3DTinyConfig(PretrainedConfig):
    model_type = "swin3d-tiny"

    def __init__(self, weights: str = "DEFAULT", **kwargs):
        super().__init__(**kwargs)
        self.weights = weights