Update README.md
Browse files
README.md
CHANGED
@@ -61,7 +61,7 @@ tokenizer = AutoTokenizer.from_pretrained('itdainb/vietnamese-cross-encoder')
|
|
61 |
|
62 |
activation_fct = torch.nn.Identity()
|
63 |
|
64 |
-
features = tokenizer(*tokenized_pairs, padding=True, truncation="longest_first", return_tensors="pt", max_length=tokenizer.config.
|
65 |
|
66 |
model.eval()
|
67 |
with torch.no_grad():
|
|
|
61 |
|
62 |
activation_fct = torch.nn.Identity()
|
63 |
|
64 |
+
features = tokenizer(*tokenized_pairs, padding=True, truncation="longest_first", return_tensors="pt", max_length=tokenizer.config.model_max_length)
|
65 |
|
66 |
model.eval()
|
67 |
with torch.no_grad():
|