sigridjineth commited on
Commit
470db40
·
verified ·
1 Parent(s): 4b65b64

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = "your-username/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,7 +99,7 @@ for s, d in ranked:
99
  import torch
100
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
101
 
102
- MODEL_ID = "your-username/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
 
 
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