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