diff --git a/lollms/utilities.py b/lollms/utilities.py index 6fef43b..c31ba8c 100644 --- a/lollms/utilities.py +++ b/lollms/utilities.py @@ -180,7 +180,7 @@ def check_torch_version(min_version, min_cuda_versio=12): def reinstall_pytorch_with_cuda(): try: - ASCIIColors.info("Installing cuda 12.1.1") + ASCIIColors.info("Installing cuda 12.1.1") # -c nvidia/label/cuda-12.1.1 -c nvidia -c conda-forge result = subprocess.run(["conda", "install", "-c", "nvidia/label/cuda-12.1.1", "cuda-toolkit", "-y"]) except Exception as ex: ASCIIColors.error(ex)