Skip to content

Commit ac8536b

Browse files
committed
Added build time warnings for llama-cpp-python package and updated dependencies in pyproject.toml and requirements.txt
1 parent 86397a1 commit ac8536b

File tree

3 files changed

+8
-2
lines changed

3 files changed

+8
-2
lines changed

install_init.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,7 @@ def install_package(package_name, extra_args=None):
106106
command = [sys.executable, "-m", "pip", "install", package_name, "--no-cache-dir"]
107107
if extra_args:
108108
command.extend(extra_args.split())
109+
print("WARNING: Building the llama-cpp-python wheels can take UP TO AN HOUR due to the specific build conditions. This is not an error, just wait.")
109110
subprocess.check_call(command)
110111

111112
def package_is_installed(package_name):
@@ -154,7 +155,6 @@ def install_llama(system_info):
154155

155156
# Build from source with appropriate acceleration
156157
try:
157-
print("WARNING: Building the llama-cpp-python wheels can take UP TO AN HOUR due to the specific build conditions. This is not an error, just wait.")
158158
if system_info.get('metal', False):
159159
print("Building llama-cpp-python from source with Metal support")
160160
os.environ['CMAKE_ARGS'] = "-DGGML_METAL=on"

pyproject.toml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,10 @@ dependencies = [
1616
"timm>=0.4.12",
1717
"tqdm>=4.65.0",
1818
"fairscale>=0.4.4",
19-
"rembg>=2.0.66" ]
19+
"rembg>=2.0.66",
20+
"diskcache>=5.6.1",
21+
"jinja2>=2.11.3",
22+
"typing-extensions>=4.5.0" ]
2023

2124
[project.urls]
2225
Repository = "https://github.com/LevelPixel/ComfyUI-LevelPixel-Advanced"

requirements.txt

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,4 +12,7 @@ tqdm>=4.65.0
1212
fairscale>=0.4.4
1313
rembg>=2.0.66
1414
matplotlib
15+
diskcache>=5.6.1
16+
jinja2>=2.11.3
17+
typing-extensions>=4.5.0
1518
ram @ git+https://github.com/xinyu1205/recognize-anything.git

0 commit comments

Comments
 (0)