wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-09 21:12:36 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Small cleanups.

Browse Source
This commit is contained in:
comfyanonymous 2023-08-20 14:56:47 -04:00
parent a670a3f848
commit 37a6cb2649
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_management.py
View File

@ -589,7 +589,10 @@ def should_use_fp16(device=None, model_params=0):
if cpu_mode() or mps_mode():
return False #TODO ?
if torch.cuda.is_bf16_supported() or xpu_available:
if xpu_available:
return True
if torch.cuda.is_bf16_supported():
return True
props = torch.cuda.get_device_properties("cuda")