File size: 259 Bytes
db45d00 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from transformers import BartConfig
class SVDCompressedBartConfig(BartConfig):
def __init__(self, *args, rank: int = 512, **kwargs):
super().__init__(*args, **kwargs)
self.rank = rank
SVDCompressedBartConfig.register_for_auto_class()
|