reverted to using embedding class

This commit is contained in:
stuart committed 2025-12-13 15:19:44 +01:00
1 parent 2473c9895f
commit bde97648e6
1 file changed
+1 -1
+1 -1
View File
@@ -36,7 +36,7 @@ class EmbeddingRequest(ModifiedBaseModel):
word_ids: List[int]
class EmbeddingResponse(ModifiedBaseModel):
embeddings: List[Tensor[torch.Tensor, tuple, Literal["float32", "float64"]]]
embeddings: List[Embedding]
class ResponseRequest(ModifiedBaseModel):
text: str