mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-11 14:50:49 +08:00
Update cuda_malloc.py
This commit is contained in:
parent
7bd86b0896
commit
66720813ff
@ -60,7 +60,8 @@ def cuda_malloc_supported():
|
||||
for b in blacklist:
|
||||
if b in x:
|
||||
return False
|
||||
return True
|
||||
return False
|
||||
#We don't need malloc at all with amd gpu's. So disabling all together.
|
||||
|
||||
|
||||
if not args.cuda_malloc:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user