This commit is contained in:
Disty0
2024-10-13 14:24:55 +03:00
parent 0c54c235cb
commit 84f8ab4076
+1 -1
View File
@@ -12,7 +12,7 @@ debug = os.environ.get('SD_DEVICE_DEBUG', None) is not None
install_traceback() # traceback handler
opts = None # initialized in get_backend to avoid circular import
args = None # initialized in get_backend to avoid circular import
cuda_ok = torch.cuda.is_available()
cuda_ok = torch.cuda.is_available() or (hasattr(torch, 'xpu') and torch.xpu.is_available())
inference_context = torch.no_grad
cpu = torch.device("cpu")