From 84ee471936b332bc2ccee024d6dfdedab4f0dc7b Mon Sep 17 00:00:00 2001
From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com>
Date: Fri, 26 Jan 2024 18:11:09 +0800
Subject: [PATCH] Add files via upload

---
 config.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/config.py b/config.py
index 897f53c..a95daf9 100644
--- a/config.py
+++ b/config.py
@@ -29,6 +29,19 @@ webui_port_subfix = 9871
 
 api_port = 9880
 
+gpu_name = torch.cuda.get_device_name(0)
+if (
+        ("16" in gpu_name and "V100" not in gpu_name.upper())
+        or "P40" in gpu_name.upper()
+        or "P10" in gpu_name.upper()
+        or "1060" in gpu_name
+        or "1070" in gpu_name
+        or "1080" in gpu_name
+):
+    is_half=False
+
+if(is_half==False and infer_device=="cuda"):infer_device="cpu"
+
 
 class Config:
     def __init__(self):