File size: 209 Bytes
a0cd7ee
 
 
 
 
 
 
1
2
3
4
5
6
7
8
from transformers import GPTJConfig


class GPTJLoraConfig(GPTJConfig):
    def __init__(self, add_adapters=False, **kwargs):
        self.add_apapters = add_adapters
        super().__init__(**kwargs)