File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
sentence_transformers/models Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -19,7 +19,7 @@ class StaticEmbedding(nn.Module):
19
19
def __init__ (
20
20
self ,
21
21
tokenizer : Tokenizer | PreTrainedTokenizerFast ,
22
- embedding_weights : np .array | torch .Tensor | None = None ,
22
+ embedding_weights : np .ndarray | torch .Tensor | None = None ,
23
23
embedding_dim : int | None = None ,
24
24
** kwargs ,
25
25
) -> None :
@@ -30,7 +30,7 @@ def __init__(
30
30
Args:
31
31
tokenizer (Tokenizer | PreTrainedTokenizerFast): The tokenizer to be used. Must be a fast tokenizer
32
32
from ``transformers`` or ``tokenizers``.
33
- embedding_weights (np.array | torch.Tensor | None, optional): Pre-trained embedding weights.
33
+ embedding_weights (np.ndarray | torch.Tensor | None, optional): Pre-trained embedding weights.
34
34
Defaults to None.
35
35
embedding_dim (int | None, optional): Dimension of the embeddings. Required if embedding_weights
36
36
is not provided. Defaults to None.
You can’t perform that action at this time.
0 commit comments