Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-21 23:57:58 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix locally compiled llama-cpp-python failing to import

Browse Source
This commit is contained in:
oobabooga 2024-10-14 13:24:13 -07:00
parent c9a9f63d1b
commit bb62e796eb
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
modules/llama_cpp_python_hijack.py
View File

@ -9,10 +9,11 @@ from modules import shared
from modules.cache_utils import process_llamacpp_cache from modules.cache_utils import process_llamacpp_cache
imported_module = None imported_module = None
not_available_modules = set()
def llama_cpp_lib(): def llama_cpp_lib():
global imported_module global imported_module, not_available_modules
# Determine the platform # Determine the platform
is_macos = platform.system() == 'Darwin' is_macos = platform.system() == 'Darwin'
@ -31,6 +32,9 @@ def llama_cpp_lib():
] ]
for arg, lib_name in lib_names: for arg, lib_name in lib_names:
if lib_name in not_available_modules:
continue
should_import = (arg is None or getattr(shared.args, arg)) should_import = (arg is None or getattr(shared.args, arg))
if should_import: if should_import:
@ -44,6 +48,7 @@ def llama_cpp_lib():
monkey_patch_llama_cpp_python(return_lib) monkey_patch_llama_cpp_python(return_lib)
return return_lib return return_lib
except ImportError: except ImportError:
not_available_modules.add(lib_name)
continue continue
return None return None