from transformers import PretrainedConfig

class ConditionalUNetConfig(PretrainedConfig):
    model_type = "conditional-unet"

    def __init__(
        self,
        encoder_rep="evanrsl/resnet-Alzheimer",
        **kwargs
    ):
        super().__init__(**kwargs)
        self.encoder_rep = encoder_rep