Better custom library debugging

Mug · Mug · commit cf339c9b3c4b · 2023-04-10T17:06:58.000+02:00
diff --git 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)

Original file line number	Diff line number	Diff line change
`@@ -26,6 +26,7 @@ def _load_shared_library(lib_base_name):`
`26`	`26`	`]`
`27`	`27`
`28`	`28`	`if ("LLAMA_LIB" in os.environ):`
	`29`	`+ lib_base_name = os.environ["LLAMA_LIB"]`
`29`	`30`	`_lib_paths = [pathlib.Path(os.environ["LLAMA_LIB"]).resolve()]`
`30`	`31`
`31`	`32`	`# Add the library directory to the DLL search path on Windows (if needed)`