From a6a53f723150683199448cfe1ee9b596d2447565 Mon Sep 17 00:00:00 2001
From: __kaning123__ <131787647+kaning123@users.noreply.github.com>
Date: Tue, 24 Feb 2026 07:48:12 +0800
Subject: [PATCH] feat: Added entry to disable checks

---
 config.json |  5 +++++
 config.py   | 14 ++++++++++++--
 2 files changed, 17 insertions(+), 2 deletions(-)
 create mode 100644 config.json

diff --git a/config.json b/config.json
new file mode 100644
index 00000000..09a5d85e
--- /dev/null
+++ b/config.json
@@ -0,0 +1,5 @@
+{
+    "GPU_CHECK":{
+        "DisableGPUMemCheck":false
+    }
+}
\ No newline at end of file
diff --git a/config.py b/config.py
index fdc11c0a..76965027 100644
--- a/config.py
+++ b/config.py
@@ -1,11 +1,20 @@
 import os
 import re
 import sys
+import json
+from pathlib import Path
 
 import torch
 
 from tools.i18n.i18n import I18nAuto
 
+current_dir = str(Path(__file__).parent)
+def merge_dir_txt2(*TXT):
+    return Path(os.path.join(*TXT))
+config_json_location = merge_dir_txt2(current_dir,"config.json")
+with open(str(config_json_location),"r") as f:
+    __info__ = f.read()
+
 i18n = I18nAuto(language=os.environ.get("language", "Auto"))
 
 
@@ -159,8 +168,9 @@ def get_device_dtype_sm(idx: int) -> tuple[torch.device, torch.dtype, float, flo
     major, minor = capability
     sm_version = major + minor / 10.0
     is_16_series = bool(re.search(r"16\d{2}", name)) and sm_version == 7.5
-    if mem_gb < 4 or sm_version < 5.3:
-        return cpu, torch.float32, 0.0, 0.0
+    if not __info__["GPU_CHECK"]["DisableGPUMemCheck"]:
+        if mem_gb < 4 or sm_version < 5.3:
+            return cpu, torch.float32, 0.0, 0.0
     if sm_version == 6.1 or is_16_series == True:
         return cuda, torch.float32, sm_version, mem_gb
     if sm_version > 6.1: