change so embeddingresponse returns list of tensors
This commit is contained in:
1 parent
f443d56882
commit
2d26ff0445
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["float32", "float64"]]
|
||||
embeddings: List[Tensor[torch.Tensor, tuple, Literal["float32", "float64"]]]
|
||||
|
||||
class ResponseRequest(ModifiedBaseModel):
|
||||
text: str
|
||||
|
||||
Reference in new issue
Block a user