changed types in embeddingresponse

This commit is contained in:
stuart committed 2025-12-13 14:44:21 +01:00
1 parent 3164e2fde7
commit f443d56882
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["int32", "int64"]]
embeddings: Tensor[torch.Tensor, tuple, Literal["float32", "float64"]]
class ResponseRequest(ModifiedBaseModel):
text: str