File size: 433 Bytes
17ff0d8
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
"""Adapted Roberta configuration for diffusion models."""

from transformers.models.roberta.configuration_roberta import RobertaConfig

from sdlm.models.mixins.configuration_mixin import DiffusionConfigMixin


class RobertaDiffusionConfig(DiffusionConfigMixin, RobertaConfig):
    def __init__(self, *args, **kwargs):
        RobertaConfig.__init__(self, *args, **kwargs)
        DiffusionConfigMixin.__init__(self, *args, **kwargs)