From b0d76185928f880fa2cde27f8614b05f6e203f9d Mon Sep 17 00:00:00 2001
From: clore <support@clore.ai>
Date: Sun, 12 May 2024 11:20:34 +0000
Subject: [PATCH] fix

---
 lib/get_specs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/get_specs.py b/lib/get_specs.py
index 0014a67..7d5c622 100644
--- a/lib/get_specs.py
+++ b/lib/get_specs.py
@@ -296,7 +296,7 @@ class Specs:
         gpu_str, gpu_mem, gpus, nvml_err = get_gpu_info()
         if require_same_gpus:
             last_gpu_name=''
-            for gpu in gpus:
+            for gpu in gpus["nvidia"]:
                 if not last_gpu_name:
                     last_gpu_name=gpu["name"]
                 elif last_gpu_name!=gpu["name"]: