Update README.md

README.md

@@ -49,7 +49,7 @@ nDCG and Recall scores of this model(out-of-domain predictions) and other multil
 alt="drawing"/>
 
 ## Usage
-First install required libraries (Flash Attention supporting GPU is a must for consistency):
+First install required libraries (Flash Attention 2 supporting GPU is a must for consistency otherwise you need to mask query expansion tokens in the output layer manually):
 
 ```bash
 pip install -U einops flash_attn
@@ -71,7 +71,7 @@ from pylate import indexes, models, retrieve
 
 # Step 1: Load the ColBERT model
 model = models.ColBERT(
-    model_name_or_path=pylate_model_id,
+    model_name_or_path="99eren99/TrColBERT",
 )
 try:
     model.tokenizer.model_input_names.remove("token_type_ids")