change so embeddingresponse returns list of tensors

This commit is contained in:
stuart committed 2025-12-13 14:48:22 +01:00
1 parent f443d56882
commit 2d26ff0445
1 file changed
+1 -1
+1 -1
View File
@@ -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