diff --git a/anansyTypes/schemas.py b/anansyTypes/schemas.py index b86d90a..b5beaa0 100644 --- a/anansyTypes/schemas.py +++ b/anansyTypes/schemas.py @@ -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