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