From e3e2ad0f37e589251d2cd9d29eb02748ce1da19b Mon Sep 17 00:00:00 2001
From: Jeevitha Kannan K S <ksjeevithakannan123@gmail.com>
Date: Sat, 12 Oct 2024 15:18:24 +0530
Subject: [PATCH] Update model fetching

---
 core/tabs/system-setup/arch/nvidia-drivers.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/core/tabs/system-setup/arch/nvidia-drivers.sh b/core/tabs/system-setup/arch/nvidia-drivers.sh
index 52cc6fe5..3c8b7c09 100755
--- a/core/tabs/system-setup/arch/nvidia-drivers.sh
+++ b/core/tabs/system-setup/arch/nvidia-drivers.sh
@@ -18,7 +18,7 @@ installDeps() {
 
 checkHardware() {
     # Refer https://nouveau.freedesktop.org/CodeNames.html for model code names
-    model=$(lspci -k | grep -A 2 -E "(VGA|3D)" | grep controller | cut -d ' ' -f 7 |  cut -c 1-2 )
+    model=$(lspci -k | grep -A 2 -E "(VGA|3D)" | grep NVIDIA | sed 's/.*Corporation //;s/ .*//' | cut -c 1-2)
     case "$model" in
         GM|GP|GV) return 1 ;;
         TU|GA|AD) return 0 ;;