From 15f99b1b710aced1ce8db70748a2a82602457661 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Mon, 18 Aug 2025 05:51:04 -0700 Subject: [PATCH] Installer: Fix a requirement file --- one_click.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/one_click.py b/one_click.py index 1ea5e59f..2c7c2c0c 100644 --- a/one_click.py +++ b/one_click.py @@ -171,14 +171,14 @@ def get_requirements_file(gpu_choice): """Get requirements file path based on GPU choice""" requirements_base = os.path.join("requirements", "full") - if gpu_choice == "AMD": + if gpu_choice == "NVIDIA_CUDA128": + file_name = f"requirements{'_noavx2' if not cpu_has_avx2() else ''}.txt" + elif gpu_choice == "AMD": file_name = f"requirements_amd{'_noavx2' if not cpu_has_avx2() else ''}.txt" elif gpu_choice == "APPLE": file_name = f"requirements_apple_{'intel' if is_x86_64() else 'silicon'}.txt" elif gpu_choice in ["INTEL", "NONE"]: file_name = f"requirements_cpu_only{'_noavx2' if not cpu_has_avx2() else ''}.txt" - elif gpu_choice == "NVIDIA_CUDA128": - file_name = f"requirements_cuda128{'_noavx2' if not cpu_has_avx2() else ''}.txt" else: raise ValueError(f"Unknown GPU choice: {gpu_choice}")