revert

lshqqytiger · Jul 7, 2024 · ee49046 · ee49046
1 parent 845d859
commit ee49046
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modules/zluda.py b/modules/zluda.py
@@ -34,7 +34,7 @@ def test(device: DeviceLikeType) -> Union[Exception, None]:
 
 def initialize_zluda():
     device = devices.get_optimal_device()
-    if not devices.cuda_ok or not is_zluda(device):
+    if not torch.cuda.is_available() or not is_zluda(device):
         return
 
     do_hijack()