getsimilarwordrequest now uses tensor
This commit is contained in:
1 parent
2d26ff0445
commit
2473c9895f
1 file changed
+1
-1
@@ -64,7 +64,7 @@ class GetWordResponse(ModifiedBaseModel):
|
||||
word: Union[dict, None]
|
||||
|
||||
class GetSimilarWordRequest(ModifiedBaseModel):
|
||||
embedding: dict
|
||||
embedding: Tensor[torch.Tensor, tuple, Literal["float32", "float64"]]
|
||||
|
||||
class GetAllPhrasesResponse(ModifiedBaseModel):
|
||||
phrases: List[dict]
|
||||
|
||||
Reference in new issue
Block a user