ukaserge
diff --git a/‎llama_cpp/llama.py
Copy file name to clipboardExpand all lines: llama_cpp/llama.py
+6-7Lines changed: 6 additions & 7 deletions b/‎llama_cpp/llama.py
Copy file name to clipboardExpand all lines: llama_cpp/llama.py
+6-7Lines changed: 6 additions & 7 deletions
@@ -176,7 +176,6 @@ def __init__(
 
         if self.verbose:
             print(llama_cpp.llama_print_system_info().decode("utf-8"), file=sys.stderr)
-        
 
         n_vocab = self.n_vocab()
         n_ctx = self.n_ctx()
@@ -575,9 +574,9 @@ def create_embedding(
         else:
             inputs = input
 
-        data = []
+        data: List[EmbeddingData] = []
         total_tokens = 0
-        for input in inputs:
+        for index, input in enumerate(inputs):
             tokens = self.tokenize(input.encode("utf-8"))
             self.reset()
             self.eval(tokens)
@@ -587,20 +586,20 @@ def create_embedding(
                 : llama_cpp.llama_n_embd(self.ctx)
             ]
 
-            if self.verbose:
-                llama_cpp.llama_print_timings(self.ctx)
             data.append(
                 {
                     "object": "embedding",
                     "embedding": embedding,
-                    "index": 0,
+                    "index": index,
                 }
             )
+        if self.verbose:
+            llama_cpp.llama_print_timings(self.ctx)
 
         return {
             "object": "list",
             "data": data,
-            "model": self.model_path,
+            "model": model_name,
             "usage": {
                 "prompt_tokens": total_tokens,
                 "total_tokens": total_tokens,