AI/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2024-10-01 01:06:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

load model library when using pyinstaller to build executable file in windows

Browse Source 
Signed-off-by: nathan <nathan@appflowy.io>
This commit is contained in:
nathan 2024-07-03 12:04:52 +08:00
parent 1ce65e8ef2
commit ce45efe605
1 changed files with 28 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
gpt4all-bindings/python/gpt4all/_pyllmodel.py
View File

@ -5,6 +5,7 @@ import os
import platform import platform
import re import re
import subprocess import subprocess
from pathlib import Path
import sys import sys
import threading import threading
from enum import Enum from enum import Enum
@ -54,16 +55,37 @@ MODEL_LIB_PATH = importlib_resources.files("gpt4all") / "llmodel_DO_NOT_MODIFY"
def load_llmodel_library(): def load_llmodel_library():
"""
Loads the llmodel shared library based on the current operating system.
This function attempts to load the shared library using the appropriate file
extension for the operating system. It first tries to load the library with the
'lib' prefix (common for macOS, Linux, and MinGW on Windows). If the file is not
found and the operating system is Windows, it attempts to load the library without
the 'lib' prefix (common for MSVC on Windows).
Returns:
ctypes.CDLL: The loaded shared library.
Raises:
OSError: If the shared library cannot be found.
"""
# Determine the appropriate file extension for the shared library based on the platform
ext = {"Darwin": "dylib", "Linux": "so", "Windows": "dll"}[platform.system()] ext = {"Darwin": "dylib", "Linux": "so", "Windows": "dll"}[platform.system()]
# Define library names with and without the 'lib' prefix
library_name_with_lib_prefix = f"libllmodel.{ext}"
library_name_without_lib_prefix = "llmodel.dll"
base_path = MODEL_LIB_PATH
try: try:
# macOS, Linux, MinGW # Attempt to load the shared library with the 'lib' prefix (common for macOS, Linux, and MinGW)
lib = ctypes.CDLL(str(MODEL_LIB_PATH / f"libllmodel.{ext}")) lib = ctypes.CDLL(str(base_path / library_name_with_lib_prefix))
except FileNotFoundError: except OSError: # OSError is more general and includes FileNotFoundError
if ext != 'dll': if ext != "dll":
raise raise
# MSVC # For Windows (ext == 'dll'), attempt to load the shared library without the 'lib' prefix (common for MSVC)
lib = ctypes.CDLL(str(MODEL_LIB_PATH / "llmodel.dll")) lib = ctypes.CDLL(str(base_path / library_name_without_lib_prefix))
return lib return lib