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 |
|