Update README.md

README.md

@@ -2684,7 +2684,7 @@ print(embeddings)
 
 ### Transformers
 
-```python
+```diff
 import torch
 import torch.nn.functional as F
 from transformers import AutoTokenizer, AutoModel
@@ -2702,10 +2702,13 @@ model.eval()
 
 encoded_input = tokenizer(sentences, padding=True, truncation=True, return_tensors='pt')
 
++ matryoshka_dim = 512
+
 with torch.no_grad():
     model_output = model(**encoded_input)
 
 embeddings = mean_pooling(model_output, encoded_input['attention_mask'])
++ embeddings = embeddings[:, :matryoshka_dim]
 embeddings = F.normalize(embeddings, p=2, dim=1)
 print(embeddings)
 ```