From 2a9951f84f27de9959c1305f8c61d316ef9b5d85 Mon Sep 17 00:00:00 2001 From: Pikurrot Date: Thu, 11 Jul 2024 22:48:35 +0200 Subject: [PATCH] Change false bool to string --- main.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index fa06cb3..f04ec6f 100644 --- a/main.py +++ b/main.py @@ -26,7 +26,10 @@ def enablePrint(): # ensure gpu_support has correct value gpu_support, error = read_config_value("gpu_support") -if error or gpu_support not in (False, "cuda", "rocm"): +if gpu_support is False: + write_config_value("gpu_support", "false") + gpu_support = "false" +if error or gpu_support not in ("false", "cuda", "rocm"): result = subprocess.run(["nvidia-smi"], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) if result.returncode == 0: write_config_value("gpu_support", "cuda")