From bef08129bce8a969582a0132c282ab1eb47cfaa4 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Wed, 6 Mar 2024 21:43:11 -0800 Subject: [PATCH] Small fix for cuda 11.8 in the one-click installer --- one_click.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/one_click.py b/one_click.py index d266be18..0d543e30 100644 --- a/one_click.py +++ b/one_click.py @@ -390,7 +390,7 @@ def update_requirements(initial_installation=False, pull=True): if is_cuda118: textgen_requirements = [req.replace('+cu121', '+cu118').replace('+cu122', '+cu118') for req in textgen_requirements] if is_windows() and is_cuda118: # No flash-attention on Windows for CUDA 11 - textgen_requirements = [req for req in textgen_requirements if 'jllllll/flash-attention' not in req] + textgen_requirements = [req for req in textgen_requirements if 'oobabooga/flash-attention' not in req] with open('temp_requirements.txt', 'w') as file: file.write('\n'.join(textgen_requirements))