Views
No views yet
[SEP] token, but token
type ids are all 0-s.1from transformers import AutoTokenizer, DPRContextEncoder
2
3tokenizer = AutoTokenizer.from_pretrained("tau/spider")
4model = DPRContextEncoder.from_pretrained("tau/spider")
5
6input_dict = tokenizer("title", "text", return_tensors="pt")
7del input_dict["token_type_ids"]
8
9outputs = model(**input_dict)