Commit
·
11ad6a6
1
Parent(s):
9bd7fbb
fix: delete prompt names besides retrieval.query
Browse files- config_sentence_transformers.json +1 -5
- custom_st.py +1 -1
config_sentence_transformers.json
CHANGED
|
@@ -5,11 +5,7 @@
|
|
| 5 |
"pytorch": "2.2.2"
|
| 6 |
},
|
| 7 |
"prompts":{
|
| 8 |
-
"retrieval.query":"Represent the query for retrieving evidence documents: "
|
| 9 |
-
"retrieval.passage":"",
|
| 10 |
-
"separation": "",
|
| 11 |
-
"classification": "",
|
| 12 |
-
"text-matching": ""
|
| 13 |
},
|
| 14 |
"default_prompt_name": null,
|
| 15 |
"similarity_fn_name": "cosine"
|
|
|
|
| 5 |
"pytorch": "2.2.2"
|
| 6 |
},
|
| 7 |
"prompts":{
|
| 8 |
+
"retrieval.query":"Represent the query for retrieving evidence documents: "
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
},
|
| 10 |
"default_prompt_name": null,
|
| 11 |
"similarity_fn_name": "cosine"
|
custom_st.py
CHANGED
|
@@ -22,7 +22,7 @@ class Transformer(nn.Module):
|
|
| 22 |
model_args: Optional[Dict[str, Any]] = None,
|
| 23 |
tokenizer_args: Optional[Dict[str, Any]] = None,
|
| 24 |
image_processor_args: Optional[Dict[str, Any]] = None,
|
| 25 |
-
assume_text_inputs: bool =
|
| 26 |
cache_dir: Optional[str] = None,
|
| 27 |
backend: Literal['torch', 'onnx', 'openvino'] = 'torch',
|
| 28 |
**_,
|
|
|
|
| 22 |
model_args: Optional[Dict[str, Any]] = None,
|
| 23 |
tokenizer_args: Optional[Dict[str, Any]] = None,
|
| 24 |
image_processor_args: Optional[Dict[str, Any]] = None,
|
| 25 |
+
assume_text_inputs: bool = False,
|
| 26 |
cache_dir: Optional[str] = None,
|
| 27 |
backend: Literal['torch', 'onnx', 'openvino'] = 'torch',
|
| 28 |
**_,
|