tomaarsen HF staff commited on
Commit
a9c5552
1 Parent(s): 981b093

Add missing space preceding comment

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -99,7 +99,7 @@ from transformers import AutoTokenizer, AutoModel
99
  import torch
100
 
101
 
102
- #Mean Pooling - Take attention mask into account for correct averaging
103
  def mean_pooling(model_output, attention_mask):
104
  token_embeddings = model_output[0] #First element of model_output contains all token embeddings
105
  input_mask_expanded = attention_mask.unsqueeze(-1).expand(token_embeddings.size()).float()
 
99
  import torch
100
 
101
 
102
+ # Mean Pooling - Take attention mask into account for correct averaging
103
  def mean_pooling(model_output, attention_mask):
104
  token_embeddings = model_output[0] #First element of model_output contains all token embeddings
105
  input_mask_expanded = attention_mask.unsqueeze(-1).expand(token_embeddings.size()).float()