Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
TerryT9 committed Dec 10, 2024
1 parent 2ae6828 commit 418d077
Show file tree
Hide file tree
Showing 4 changed files with 4 additions and 5 deletions.
3 changes: 1 addition & 2 deletions nexa/gguf/lib_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,7 @@ def is_gpu_available():
# Load the library
def load_library(lib_base_name: str):
# Construct the paths to the possible shared library names
# _base_path = pathlib.Path(os.path.abspath(os.path.dirname(__file__))) / "lib"
_base_path = pathlib.Path('D:/repo/nexa-ai/llama-cpp-python/llama_cpp/lib')
_base_path = pathlib.Path(os.path.abspath(os.path.dirname(__file__))) / "lib"
# Searching for the library in the current directory under the name "libllama" (default name
# for llamacpp) and "llama" (default name for this repo)
_lib_paths: List[pathlib.Path] = []
Expand Down
2 changes: 1 addition & 1 deletion nexa/gguf/llama/llama_chat_format.py
Original file line number Diff line number Diff line change
Expand Up @@ -2667,7 +2667,7 @@ class Llava15ChatHandler:
)

def __init__(self, clip_model_path: str, verbose: bool = True):
import llama_cpp.llava_cpp as llava_cpp
import nexa.gguf.llama.llava_cpp as llava_cpp

self.clip_model_path = clip_model_path
self.verbose = verbose
Expand Down
2 changes: 1 addition & 1 deletion nexa/gguf/llama/llava_cpp.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@


# Specify the base name of the shared library to load
_libllava_base_name = "llava"
_libllava_base_name = "llava_shared"
# Load the library
_libllava = load_library(_libllava_base_name)

Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -131,7 +131,7 @@ cmake.args = [
"-DCMAKE_BUILD_PARALLEL_LEVEL=16",
"-DSTABLE_DIFFUSION_BUILD=ON",
"-DLLAMA_BUILD=ON",
"-DBARK_BUILD=ON",
"-DBARK_BUILD=OFF",
"-DBUILD_SHARED_LIBS=ON",
"-DLLAMA_BUILD_TESTS=OFF"
]
Expand Down

0 comments on commit 418d077

Please sign in to comment.