Spaces:
Running
Running
Germano Cavalcante
commited on
Commit
•
15d912e
1
Parent(s):
66906e4
Falback for GPU checker
Browse files
routers/tool_gpu_checker.py
CHANGED
@@ -176,6 +176,13 @@ def gpu_checker_get_message(text):
|
|
176 |
vendor = 'Apple'
|
177 |
is_supported, model, descr = _check_apple(text)
|
178 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
179 |
if not vendor:
|
180 |
return "Could not find graphics card information"
|
181 |
elif not model:
|
|
|
176 |
vendor = 'Apple'
|
177 |
is_supported, model, descr = _check_apple(text)
|
178 |
|
179 |
+
else:
|
180 |
+
for func in {_check_nvidia, _check_amd, _check_intel, _check_apple}:
|
181 |
+
is_supported, model, descr = func(text)
|
182 |
+
if model:
|
183 |
+
vendor = 'GPU'
|
184 |
+
break
|
185 |
+
|
186 |
if not vendor:
|
187 |
return "Could not find graphics card information"
|
188 |
elif not model:
|