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()