diff --git a/custom_components/llama_conversation/utils.py b/custom_components/llama_conversation/utils.py index 1b9281d..d3b4ccd 100644 --- a/custom_components/llama_conversation/utils.py +++ b/custom_components/llama_conversation/utils.py @@ -166,13 +166,16 @@ def install_llama_cpp_python(config_dir: str): return True platform_suffix = platform.machine() + # remap other names for architectures to the names we use if platform_suffix == "arm64": platform_suffix = "aarch64" + if platform_suffix == "i386" or platform_suffix == "x86_64": + platform_suffix = "amd64" runtime_version = f"cp{sys.version_info.major}{sys.version_info.minor}" instruction_extensions_suffix = "" - if platform_suffix == "amd64" or platform_suffix == "i386": + if platform_suffix == "amd64": instruction_extensions_suffix = "-noavx" try: