Spaces:
Sleeping
Sleeping
Update RoBERTaModule.py
Browse files- 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 |
|