PeteBleackley commited on
Commit
2f6dc26
·
1 Parent(s): 1d631bc

Problems with config

Browse files
Files changed (1) hide show
  1. qarac/models/QaracBaseModel.py +1 -1
qarac/models/QaracBaseModel.py CHANGED
@@ -10,7 +10,7 @@ import transformers
10
 
11
  class QaracBaseModel(transformers.PreTrainedModel):
12
  """Base class for Qarac Models. Provided config_class"""
13
- config_class = transformers.RobertaConfig
14
 
15
  def __init__(self,config,*inputs,**kwargs):
16
  super(QaracBaseModel,self).__init__(config,*inputs,**kwargs)
 
10
 
11
  class QaracBaseModel(transformers.PreTrainedModel):
12
  """Base class for Qarac Models. Provided config_class"""
13
+ config_class = transformers.PretrainedConfig
14
 
15
  def __init__(self,config,*inputs,**kwargs):
16
  super(QaracBaseModel,self).__init__(config,*inputs,**kwargs)