itdainb commited on
Commit
4d1059e
·
verified ·
1 Parent(s): c212b98

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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.max_length)
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():