File size: 304 Bytes
dd983c1 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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 |