Better custom library debugging

2023-04-10 17:06:58 +02:00 · 2023-04-10 17:06:58 +02:00 · cf339c9b3c
parent 4132293d2d
commit cf339c9b3c
1 changed files with 1 additions and 0 deletions
--- 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)