Markus28 commited on
Commit
fde927f
·
1 Parent(s): 318cbce

fix: fix typo

Browse files
Files changed (1) hide show
  1. modeling_bert.py +1 -1
modeling_bert.py CHANGED
@@ -298,7 +298,7 @@ class BertPreTrainingHeads(nn.Module):
298
  return prediction_scores, seq_relationship_score
299
 
300
 
301
- class BertPreTrainedModel(PretrainedModel): #(nn.Module):
302
  """An abstract class to handle weights initialization and
303
  a simple interface for dowloading and loading pretrained models.
304
  """
 
298
  return prediction_scores, seq_relationship_score
299
 
300
 
301
+ class BertPreTrainedModel(PreTrainedModel): #(nn.Module):
302
  """An abstract class to handle weights initialization and
303
  a simple interface for dowloading and loading pretrained models.
304
  """