fix: create_embedding broken response for input type str

2024-02-15 16:09:48 -05:00 · 2024-02-15 16:09:48 -05:00 · 0ce66bc080
commit 0ce66bc080
parent ea1f88dd29
1 changed files with 2 additions and 0 deletions
--- a/llama_cpp/llama.py
+++ b/llama_cpp/llama.py
@ -720,6 +720,8 @@ class Llama:
        assert self._model.model is not None
        model_name: str = model if model is not None else self.model_path

+        input = input if isinstance(input, list) else [input]
+
        # get numeric embeddings
        embeds: List[List[float]]
        total_tokens: int