Changes in embedding.py
Adjusted embedding data tensor shape
This commit is contained in:
1 parent
fabcdad845
commit
6782d23044
1 file changed
+1
-1
@@ -8,7 +8,7 @@ from typing import Any, Literal
|
||||
|
||||
@dataclass(frozen=True)
|
||||
class Embedding:
|
||||
data: Tensor[torch.Tensor | np.ndarray[Any, Any], tuple, Literal["int32", "int64"]]
|
||||
data: Tensor[torch.Tensor | np.ndarray[Any], tuple, Literal["int64"]]
|
||||
|
||||
def compare(self, other: Embedding) -> float:
|
||||
"""
|
||||
|
||||
Reference in new issue
Block a user