File size: 179 Bytes
137ceae
608eb1e
137ceae
 
 
1
2
3
4
5
from transformers import GPT2LMHeadModel
from .configuration_cnb3 import CaelisNeuralBaseConfig

class CaelisNeuralBase(GPT2LMHeadModel):
    config_class = CaelisNeuralBaseConfig