ollama/server/build.py

21 lines
477 B
Python
Raw Normal View History

2023-06-23 17:18:47 -04:00
import site
import os
from PyInstaller.__main__ import run as pyi_run
# Get the directory of site-packages and llama_cpp
site_packages_dir = site.getsitepackages()[0]
llama_cpp_dir = os.path.join(site_packages_dir, "llama_cpp")
# Prepare the arguments for PyInstaller
args = [
"server.py",
"--paths",
site_packages_dir,
"--add-data",
f"{llama_cpp_dir}{os.pathsep}llama_cpp",
"--onefile",
]
# Generate the .spec file and run PyInstaller
pyi_run(args)