d2527f6
1
2
3
4
5
6
7
from transformers import DebertaV2Config class MultiHeadDebertaV2Config(DebertaV2Config): model_type = "multi-head-deberta-for-sequence-classification" def __init__(self, num_heads=5, **kwargs): self.num_heads = num_heads super().__init__(**kwargs)