gmastrapas
commited on
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 |
**_,
|