gmastrapas commited on
Commit
11ad6a6
1 Parent(s): 9bd7fbb

fix: delete prompt names besides retrieval.query

Browse files
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 = True,
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
  **_,