Update README.md
Browse files
README.md
CHANGED
@@ -64,7 +64,7 @@ Below are **drop-in** examples for the converted model. These mirror the origina
|
|
64 |
```python
|
65 |
from sentence_transformers import CrossEncoder
|
66 |
|
67 |
-
MODEL_ID = "
|
68 |
|
69 |
def format_prompts(query: str, instruction: str, docs: list[str]) -> list[str]:
|
70 |
inst = f" {instruction}" if instruction else ""
|
@@ -99,7 +99,7 @@ for s, d in ranked:
|
|
99 |
import torch
|
100 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
101 |
|
102 |
-
MODEL_ID = "
|
103 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
104 |
dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
|
105 |
|
|
|
64 |
```python
|
65 |
from sentence_transformers import CrossEncoder
|
66 |
|
67 |
+
MODEL_ID = "sigridjineth/ctxl-rerank-v2-1b-seq-cls" # or local folder
|
68 |
|
69 |
def format_prompts(query: str, instruction: str, docs: list[str]) -> list[str]:
|
70 |
inst = f" {instruction}" if instruction else ""
|
|
|
99 |
import torch
|
100 |
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
101 |
|
102 |
+
MODEL_ID = "sigridjineth/ctxl-rerank-v2-1b-seq-cls" # or local folder
|
103 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
104 |
dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
|
105 |
|