Update README.md
Browse files
README.md
CHANGED
|
@@ -141,8 +141,6 @@ class CustomModel:
|
|
| 141 |
prompt_type: PromptType | None = None,
|
| 142 |
**kwargs,
|
| 143 |
) -> np.ndarray:
|
| 144 |
-
if prompt_type == PromptType.query:
|
| 145 |
-
sentences = [f"task: search result | query: {s}" for s in sentences]
|
| 146 |
inputs = self.tokenizer(sentences, padding=True, truncation=True, return_tensors="np", max_length=8192)
|
| 147 |
q = self.session.run(["sentence_embedding"], dict(inputs))
|
| 148 |
return self.dequantize(q, self.scale)
|
|
|
|
| 141 |
prompt_type: PromptType | None = None,
|
| 142 |
**kwargs,
|
| 143 |
) -> np.ndarray:
|
|
|
|
|
|
|
| 144 |
inputs = self.tokenizer(sentences, padding=True, truncation=True, return_tensors="np", max_length=8192)
|
| 145 |
q = self.session.run(["sentence_embedding"], dict(inputs))
|
| 146 |
return self.dequantize(q, self.scale)
|