Disable cuda malloc on regular GTX 960.

comfyanonymous · Jul 22, 2023 · 85a8900 · 85a8900
1 parent 78e7958
commit 85a8900
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cuda_malloc.py b/cuda_malloc.py
@@ -37,7 +37,7 @@ def enum_display_devices():
         return set()
 
 def cuda_malloc_supported():
-    blacklist = {"GeForce GTX 960M", "GeForce GTX 950M", "GeForce 945M", "GeForce 940M", "GeForce 930M", "GeForce 920M", "GeForce 910M", "GeForce GTX 750", "GeForce GTX 745"}
+    blacklist = {"GeForce GTX 960", "GeForce GTX 950", "GeForce 945M", "GeForce 940M", "GeForce 930M", "GeForce 920M", "GeForce 910M", "GeForce GTX 750", "GeForce GTX 745"}
     try:
         names = get_gpu_names()
     except: