gmastrapas commited on
Commit
9bd7fbb
1 Parent(s): 5695d43

feat: add option assume_text_inputs in sentence transformers

Browse files
Files changed (1) hide show
  1. custom_st.py +1 -1
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 = False,
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 = True,
26
  cache_dir: Optional[str] = None,
27
  backend: Literal['torch', 'onnx', 'openvino'] = 'torch',
28
  **_,