Spaces:
Sleeping
Sleeping
File size: 602 Bytes
17ff0d8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
"""Adapted Mistral configuration for diffusion models."""
from transformers.models.mistral import MistralConfig
from sdlm.models.mixins.configuration_mixin import DiffusionConfigMixin
class MistralDiffusionConfig(DiffusionConfigMixin, MistralConfig):
def __init__(self, *args, **kwargs):
MistralConfig.__init__(self, *args, **kwargs)
DiffusionConfigMixin.__init__(self, *args, **kwargs)
class CDCDMistralDiffusionConfig(MistralDiffusionConfig):
def __init__(self, *args, n_bins: int = 100, **kwargs):
super().__init__(*args, **kwargs)
self.n_bins = n_bins
|