 How to use  ```python from transformers import AutoTokenizer, AutoModel  def mean_pooling(token_embeddings, mask):     token_embeddings = token_embeddings.masked_fill(~mask[..., None].bool(), 0.)     sentence_embeddings = token_embeddings.sum(dim=1) / mask.sum(dim=1)[..., None]     return sentence_embeddings  