Germano Cavalcante commited on
Commit
15d912e
1 Parent(s): 66906e4

Falback for GPU checker

Browse files
Files changed (1) hide show
  1. routers/tool_gpu_checker.py +7 -0
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: