From a16ca5d057262d681df3bbe62a30a10e28c8782b Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Sun, 11 Aug 2024 17:53:08 -0700
Subject: [PATCH] fix amd

---
 backend/memory_management.py | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/backend/memory_management.py b/backend/memory_management.py
index 050d7ea8..3fa5e5f1 100644
--- a/backend/memory_management.py
+++ b/backend/memory_management.py
@@ -1078,15 +1078,18 @@ def should_use_bf16(device=None, model_params=0, prioritize_performance=True, ma
 
 
 def can_install_bnb():
-    if not torch.cuda.is_available():
+    try:
+        if not torch.cuda.is_available():
+            return False
+
+        cuda_version = tuple(int(x) for x in torch.version.cuda.split('.'))
+
+        if cuda_version >= (11, 7):
+            return True
+
+        return False
+    except:
         return False
-
-    cuda_version = tuple(int(x) for x in torch.version.cuda.split('.'))
-
-    if cuda_version >= (11, 7):
-        return True
-
-    return False
 
 
 def soft_empty_cache(force=False):