wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-07 07:42:32 +08:00
Code Issues Actions 18 Packages Projects Releases Wiki Activity

Don't use xformers nor fp16 in AMD GPUs

Browse Source 
Don't use xformers nor fp16 in AMD GPUs (not supported by them). Example name:

```
In [8]: torch.cuda.get_device_properties("cuda").name
Out[8]: 'AMD Radeon RX 5700 XT'
```
This commit is contained in:
Pedro Batista 2023-04-05 13:14:15 -03:00 committed by GitHub
parent 30f274bf48
commit 0169c95d82
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_management.py
View File

@ -203,6 +203,8 @@ def get_autocast_device(dev):
def xformers_enabled():
if vram_state == CPU:
return False
if torch.cuda.get_device_properties("cuda").name == "AMD":
return False
return XFORMERS_IS_AVAILBLE
@ -268,6 +270,9 @@ def should_use_fp16():
props = torch.cuda.get_device_properties("cuda")
if props.major < 7:
return False
if "AMD" in props.name:
return False
#FP32 is faster on those cards?
nvidia_16_series = ["1660", "1650", "1630", "T500", "T550", "T600"]