change from_dict from np to torch

This commit is contained in:
stuart committed 2025-12-14 00:42:44 +01:00
1 parent e4e387b3b9
commit 6ec0fac133
1 file changed
+1 -1
+1 -1
View File
@@ -25,4 +25,4 @@ class Embedding:
@classmethod
def from_dict(cls, data: dict) -> "Embedding":
return cls(data=np.array(data["data"], dtype=np.float32))
return cls(data=torch.tensor(data["data"], dtype=torch.float32))