wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-17 08:52:34 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Enable FP8 compute on AMD

Browse Source
This commit is contained in:
Your Name 2025-09-18 10:06:39 +03:00
parent dd611a7700
commit 9123ec20c3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@ -1355,7 +1355,7 @@ def supports_fp8_compute(device=None):
if SUPPORT_FP8_OPS: if SUPPORT_FP8_OPS:
return True return True
if not is_nvidia(): if not (is_nvidia() or is_amd()):
return False return False
props = torch.cuda.get_device_properties(device) props = torch.cuda.get_device_properties(device)