token_dict = tokenizer(s1, return_tensors='pt') token_ids = token_dict['input_ids'][0]