vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Bugfix: cross-platform method to find shared lib

Browse source
This commit is contained in:
Andrei Betlen 2023-03-24 18:43:29 -04:00 • committed by Don Mahurin
parent bd1c657f80
commit a3da39af79
1 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

10
examples/llama_cpp.py
View file

@ -12,11 +12,15 @@ from ctypes import (
)
import pathlib
from itertools import chain
# Load the library
libfile = pathlib.Path(__file__).parent / "libllama.so"
_lib = ctypes.CDLL(str(libfile))
# TODO: fragile, should fix
_base_path = pathlib.Path(__file__).parent
(_lib_path,) = chain(
_base_path.glob("*.so"), _base_path.glob("*.dylib"), _base_path.glob("*.dll")
)
_lib = ctypes.CDLL(str(_lib_path))
# C types
llama_context_p = c_void_p