File size: 270 Bytes
668789a |
1 2 3 4 5 6 7 8 9 10 11 |
from transformers import GPTJConfig
class GPTJLoraConfig(GPTJConfig):
model_type = "gptj-lora"
def __init__(self, add_adapters=False, **kwargs):
self.add_apapters = add_adapters
super().__init__(**kwargs)
self.model_type = "gptj-lora"
|