File size: 274 Bytes
ceadefe |
1 2 3 4 5 6 7 8 9 10 |
from transformers.models.mistral.configuration_mistral import MistralConfig
class EvoMistralConfig(MistralConfig):
model_type = "evomistral"
def __init__(self, num_hops: int = 64, **kwargs):
self.num_hops = num_hops
super().__init__(**kwargs)
|