bowphs commited on
Commit
6e26ea5
·
1 Parent(s): d693721

Update modeling_depenberta.py

Browse files
Files changed (1) hide show
  1. modeling_depenberta.py +1 -0
modeling_depenberta.py CHANGED
@@ -3,6 +3,7 @@ from transformers.modeling_outputs import TokenClassifierOutput
3
  from transformers.models.roberta.modeling_roberta import RobertaModel
4
  from transformers.models.roberta.modeling_roberta import RobertaPreTrainedModel
5
  import torch.nn as nn
 
6
 
7
 
8
  class DependencyRobertaForTokenClassification(RobertaPreTrainedModel):
 
3
  from transformers.models.roberta.modeling_roberta import RobertaModel
4
  from transformers.models.roberta.modeling_roberta import RobertaPreTrainedModel
5
  import torch.nn as nn
6
+ import torch
7
 
8
 
9
  class DependencyRobertaForTokenClassification(RobertaPreTrainedModel):