build.py 451 Bytes
Newer Older
Bruce MacDonald's avatar
Bruce MacDonald committed
1
2
3
4
import site
import os
from PyInstaller.__main__ import run as pyi_run

Bruce MacDonald's avatar
Bruce MacDonald committed
5
# the llama_cpp directory is not included if not explicitly added
Bruce MacDonald's avatar
Bruce MacDonald committed
6
7
8
9
site_packages_dir = site.getsitepackages()[0]
llama_cpp_dir = os.path.join(site_packages_dir, "llama_cpp")

args = [
Jeffrey Morgan's avatar
Jeffrey Morgan committed
10
    "ollama.py",
Bruce MacDonald's avatar
Bruce MacDonald committed
11
12
13
14
    "--paths",
    site_packages_dir,
    "--add-data",
    f"{llama_cpp_dir}{os.pathsep}llama_cpp",
Jeffrey Morgan's avatar
Jeffrey Morgan committed
15
    "--onefile"
Bruce MacDonald's avatar
Bruce MacDonald committed
16
17
]

Bruce MacDonald's avatar
Bruce MacDonald committed
18
# generate the .spec file and run PyInstaller
Bruce MacDonald's avatar
Bruce MacDonald committed
19
pyi_run(args)