Update app.py

app.py

@@ -3,7 +3,6 @@ import sys
 import site
 
 # ================= 强力 CPU 环境全面拟真补丁 =================
-# 拦截启动参数
 if "--use-sage-attention" in sys.argv:
     sys.argv.remove("--use-sage-attention")
 
@@ -21,9 +20,8 @@ try:
     torch.cuda.device_count = lambda: 0
     torch.cuda.current_device = lambda: 0
     
-    # 2. 【核心修复】伪造 GPU 内存统计数据，防止 KeyError
+    # 2. 伪造 GPU 内存统计数据与 C++ 运行时属性
     def mock_memory_stats(*args, **kwargs):
-        # 模拟一个拥有基础显存键值的字典，让 ComfyUI 读取时不崩溃
         return {
             'reserved_bytes.all.current': 0,
             'reserved_bytes.all.peak': 0,
@@ -36,6 +34,17 @@ try:
     torch.cuda.memory_reserved = lambda *args, **kwargs: 0
     torch.cuda.max_memory_reserved = lambda *args, **kwargs: 0
     
+    # 【核心修复】：拦截 mem_get_info，返回假的 (空闲显存, 总显存)，直接给它 16GB 虚拟显存
+    torch.cuda.mem_get_info = lambda *args, **kwargs: (16 * 1024 * 1024 * 1024, 16 * 1024 * 1024 * 1024)
+    
+    # 额外拦截设备属性查询，防止接下来报错
+    class MockDeviceProperties:
+        name = "Kaggle/HuggingFace CPU-Fake GPU"
+        major = 8
+        minor = 0
+        total_memory = 16 * 1024 * 1024 * 1024
+    torch.cuda.get_device_properties = lambda *args, **kwargs: MockDeviceProperties()
+
     # 3. 构造设备代理类，保持 type 身份
     _orig_device = torch.device
 
@@ -57,7 +66,7 @@ try:
     TorchDeviceProxy.__qualname__ = _orig_device.__qualname__
 
     torch.device = TorchDeviceProxy
-    print("🤖 [CPU Patch] Seamless TorchDeviceProxy and MemoryStats deployed.")
+    print("🤖 [CPU Patch] Advanced TorchCuda/MemGetInfo mock deployed.")
 except Exception as e:
     print(f"⚠️ [CPU Patch] Error while generating proxy patch: {e}")