baalajimaestro/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix: create_embedding broken response for input type str

Browse source
This commit is contained in:
Andrei Betlen 2024-02-15 16:09:48 -05:00
parent ea1f88dd29
commit 0ce66bc080
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
llama_cpp/llama.py
View file

@ -720,6 +720,8 @@ class Llama:
assert self._model.model is not None assert self._model.model is not None
model_name: str = model if model is not None else self.model_path model_name: str = model if model is not None else self.model_path
input = input if isinstance(input, list) else [input]
# get numeric embeddings # get numeric embeddings
embeds: List[List[float]] embeds: List[List[float]]
total_tokens: int total_tokens: int