From 6ecce1604b01ded0b2d1163008f3010c5f125384 Mon Sep 17 00:00:00 2001 From: kingbri Date: Sun, 31 Mar 2024 23:11:21 -0400 Subject: [PATCH] Model: Fix log if exl2 version is too low Switch to pyproject syntax. Signed-off-by: kingbri --- backends/exllamav2/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backends/exllamav2/utils.py b/backends/exllamav2/utils.py index c6b0a62..4ee0c4a 100644 --- a/backends/exllamav2/utils.py +++ b/backends/exllamav2/utils.py @@ -18,11 +18,11 @@ def check_exllama_version(): "Or you can manually run a requirements update " "using the following command:\n\n" "For CUDA 12.1:\n" - "pip install --upgrade -r requirements.txt\n\n" + "pip install --upgrade .[cu121]\n\n" "For CUDA 11.8:\n" - "pip install --upgrade -r requirements-cu118.txt\n\n" + "pip install --upgrade .[cu118]\n\n" "For ROCm:\n" - "pip install --upgrade -r requirements-amd.txt\n\n" + "pip install --upgrade .[amd]\n\n" ) else: logger.info(f"ExllamaV2 version: {current_version}")