lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 18:51:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix amd

Browse Source
This commit is contained in:
layerdiffusion
2024-08-11 17:53:08 -07:00
parent e72d912118
commit a16ca5d057
1 changed files with 11 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
backend/memory_management.py
View File

@@ -1078,15 +1078,18 @@ def should_use_bf16(device=None, model_params=0, prioritize_performance=True, ma
def can_install_bnb(): 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 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): def soft_empty_cache(force=False):