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 | 
