change from_dict from np to torch
This commit is contained in:
1 parent
e4e387b3b9
commit
6ec0fac133
1 file changed
+1
-1
@@ -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))
|
||||
Reference in new issue
Block a user