From a28cb92d8f4c39b11f2acccaaf7013a5c0e27278 Mon Sep 17 00:00:00 2001
From: Andrei Betlen <abetlen@gmail.com>
Date: Fri, 24 Mar 2023 04:04:29 -0400
Subject: [PATCH] Remove model_name param

---
 llama_cpp/llama.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/llama_cpp/llama.py b/llama_cpp/llama.py
index b57a41e..7f7abaa 100644
--- a/llama_cpp/llama.py
+++ b/llama_cpp/llama.py
@@ -16,10 +16,8 @@ class Llama:
         logits_all: bool = False,
         vocab_only: bool = False,
         n_threads: Optional[int] = None,
-        model_name: Optional[str]=None,
     ):
         self.model_path = model_path
-        self.model = model_name or model_path
 
         self.params = llama_cpp.llama_context_default_params()
         self.params.n_ctx = n_ctx
@@ -122,7 +120,7 @@ class Llama:
             "id": f"cmpl-{str(uuid.uuid4())}", # Likely to change
             "object": "text_completion",
             "created": int(time.time()),
-            "model": self.model, # Likely to change
+            "model": self.model_path,
             "choices": [
                 {
                     "text": text,