gpt-j-6b-8bit-lora / config.py
Enkhai's picture
Update config.py
38e9038
raw
history blame
No virus
219 Bytes
from transformers import GPTJConfig, AutoConfig
class GPTJLoraConfig(GPTJConfig):
def __init__(self, add_adapters=False, **kwargs):
self.add_apapters = add_adapters
super().__init__(**kwargs)