Update README.md

README.md

@@ -2704,7 +2704,7 @@ with torch.no_grad():
     model_output = model(**encoded_input)
 
 embeddings = mean_pooling(model_output, encoded_input['attention_mask'])
-+ embeddings = F.layer_norm(embeddings, normalized_shape=embeddings.shape[1], dim=-1)
++ embeddings = F.layer_norm(embeddings, normalized_shape=(embeddings.shape[1],))
 + embeddings = embeddings[:, :matryoshka_dim]
 embeddings = F.normalize(embeddings, p=2, dim=1)
 print(embeddings)