reverted to using embedding class
This commit is contained in:
1 parent
2473c9895f
commit
bde97648e6
1 file changed
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user