Update the snippet slightly

README.md

@@ -139,7 +139,7 @@ pip install sentence-transformers
 ```python
 from sentence_transformers import SentenceTransformer
 
-attn_implementation = "eager"
+attn_implementation = "eager"  # Or "flash_attention_2"
 model = SentenceTransformer(
     "nvidia/llama-embed-nemotron-8b",
     trust_remote_code=True,
@@ -155,6 +155,7 @@ documents = [
     "Market prices are determined by the relationship between how much people want to buy a product and how much is available for sale, with scarcity driving prices up and abundance driving them down.",
 ]
 
+# NOTE: encode_query uses the "query" prompt automatically
 query_embeddings = model.encode_query(queries)
 document_embeddings = model.encode_document(documents)