gmastrapas
commited on
Commit
•
9bd7fbb
1
Parent(s):
5695d43
feat: add option assume_text_inputs in sentence transformers
Browse files- 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 =
|
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 |
**_,
|