wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 14:50:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Update cuda_malloc.py

Browse Source
This commit is contained in:
patientx 2024-08-26 14:42:12 +03:00 committed by GitHub
parent 58594a0b47
commit d7f28efe8d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cuda_malloc.py
View File

@ -56,12 +56,14 @@ def cuda_malloc_supported():
except:
names = set()
for x in names:
if "NVIDIA" in x:
if "AMD" in x:
return False
elif "NVIDIA" in x:
for b in blacklist:
if b in x:
return False
return False
#We don't need malloc at all with amd gpu's. So disabling all together.
return True
#returned to older version.
if not args.cuda_malloc: