298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-03-13 14:21:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small cleanups.

Browse Source
This commit is contained in:
comfyanonymous 2023-08-20 14:56:47 -04:00
parent 9225465975
commit 0d7b0a4dc7
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")