99eren99 commited on
Commit
dd3cc59
·
verified ·
1 Parent(s): 7b8081f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -49,7 +49,7 @@ nDCG and Recall scores of this model(out-of-domain predictions) and other multil
49
  alt="drawing"/>
50
 
51
  ## Usage
52
- First install required libraries (Flash Attention 2 supporting GPU is a must for consistency otherwise you need to mask query expansion token in the output layer manually):
53
 
54
  ```bash
55
  pip install -U einops flash_attn
@@ -72,7 +72,7 @@ from pylate import indexes, models, retrieve
72
  # Step 1: Load the ColBERT model
73
  document_length = 8192 #[1,8192] for truncating documents
74
  model = models.ColBERT(
75
- model_name_or_path="99eren99/TrColbert-Long",document_length=document_length
76
  )
77
  try:
78
  model.tokenizer.model_input_names.remove("token_type_ids")
 
49
  alt="drawing"/>
50
 
51
  ## Usage
52
+ 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):
53
 
54
  ```bash
55
  pip install -U einops flash_attn
 
72
  # Step 1: Load the ColBERT model
73
  document_length = 8192 #[1,8192] for truncating documents
74
  model = models.ColBERT(
75
+ model_name_or_path="99eren99/TrColBERT-Long",document_length=document_length
76
  )
77
  try:
78
  model.tokenizer.model_input_names.remove("token_type_ids")