File size: 219 Bytes
50f0ca6
 
 
 
d9c5f3b
50f0ca6
 
 
1
2
3
4
5
6
7
8
9
from transformers import GPTJConfig, AutoConfig


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