diff --git a/examples/llama_cpp.py b/examples/llama_cpp.py index 63e8e97bf..89eca4bb7 100644 --- a/examples/llama_cpp.py +++ b/examples/llama_cpp.py @@ -37,6 +37,7 @@ def _load_shared_library(lib_base_name): ] if ("LLAMA_CPP_LIB" in os.environ): + lib_base_name = os.environ["LLAMA_CPP_LIB"] _lib_paths = [pathlib.Path(os.environ["LLAMA_CPP_LIB"]).resolve()] # Add the library directory to the DLL search path on Windows (if needed)