changed types in embeddingresponse
This commit is contained in:
1 parent
3164e2fde7
commit
f443d56882
1 file changed
+1
-1
@@ -36,7 +36,7 @@ class EmbeddingRequest(ModifiedBaseModel):
|
||||
word_ids: List[int]
|
||||
|
||||
class EmbeddingResponse(ModifiedBaseModel):
|
||||
embeddings: Tensor[torch.Tensor, tuple, Literal["int32", "int64"]]
|
||||
embeddings: Tensor[torch.Tensor, tuple, Literal["float32", "float64"]]
|
||||
|
||||
class ResponseRequest(ModifiedBaseModel):
|
||||
text: str
|
||||
|
||||
Reference in new issue
Block a user