Update modeling_xlm_roberta.py
#24
by
bwang0911
- opened
- modeling_xlm_roberta.py +0 -1
modeling_xlm_roberta.py
CHANGED
|
@@ -548,7 +548,6 @@ class XLMRobertaModel(XLMRobertaPreTrainedModel):
|
|
| 548 |
for i in range_iter:
|
| 549 |
encoded_input = self.tokenizer(
|
| 550 |
sentences[i : i + batch_size],
|
| 551 |
-
return_tensors='pt',
|
| 552 |
**tokenizer_kwargs,
|
| 553 |
).to(self.device)
|
| 554 |
token_embs = self.forward(**encoded_input)[0]
|
|
|
|
| 548 |
for i in range_iter:
|
| 549 |
encoded_input = self.tokenizer(
|
| 550 |
sentences[i : i + batch_size],
|
|
|
|
| 551 |
**tokenizer_kwargs,
|
| 552 |
).to(self.device)
|
| 553 |
token_embs = self.forward(**encoded_input)[0]
|