From cf339c9b3c4bc3588b313a07713af3dd5c3e3724 Mon Sep 17 00:00:00 2001 From: Mug <> Date: Mon, 10 Apr 2023 17:06:58 +0200 Subject: [PATCH] Better custom library debugging --- llama_cpp/llama_cpp.py | 1 + 1 file changed, 1 insertion(+) 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)