Changed the environment variable name into "LLAMA_CPP_LIB"

Mug · Mug · commit 2559e5af9b04 · 2023-04-10T17:27:17.000+02:00
diff --git a/llama_cpp/llama_cpp.py b/llama_cpp/llama_cpp.py
@@ -25,8 +25,8 @@ def _load_shared_library(lib_base_name):
         _base_path / f"{lib_base_name}{lib_ext}"
     ]
 
-    if ("LLAMA_LIB" in os.environ):
-        lib_base_name = os.environ["LLAMA_LIB"]
+    if ("LLAMA_CPP_LIB" in os.environ):
+        lib_base_name = os.environ["LLAMA_CPP_LIB"]
         _lib = pathlib.Path(lib_base_name)
         _base_path = _lib.parent.resolve()
         _lib_paths = [_lib.resolve()]
