DornierDo17 commited on
Commit
2dd717f
·
verified ·
1 Parent(s): d317fe7

Update RoBERTaModule.py

Browse files
Files changed (1) hide show
  1. RoBERTaModule.py +0 -3
RoBERTaModule.py CHANGED
@@ -11,9 +11,6 @@ class RoBERTaModule(nn.Module):
11
  self.tokenizer = RobertaTokenizerFast.from_pretrained("roberta-base")
12
  self.model = RoBERTa(vocab_size=self.tokenizer.vocab_size, padding_idx=self.tokenizer.pad_token_id)
13
 
14
- def forward(self, x, attn_mask):
15
- return self.model(x, attn_mask)
16
-
17
  def forward(self, x, attn_mask):
18
  return self.model(x, attn_mask)
19
 
 
11
  self.tokenizer = RobertaTokenizerFast.from_pretrained("roberta-base")
12
  self.model = RoBERTa(vocab_size=self.tokenizer.vocab_size, padding_idx=self.tokenizer.pad_token_id)
13
 
 
 
 
14
  def forward(self, x, attn_mask):
15
  return self.model(x, attn_mask)
16