jinx/ComfyUI
1
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00
Code Issues Activity

Fix crash.

Browse Source
This commit is contained in:
comfyanonymous
2024-08-03 16:55:38 -04:00
parent 91be9c2867
commit f7a5107784
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
comfy/model_management.py
View File

@@ -928,10 +928,7 @@ def should_use_bf16(device=None, model_params=0, prioritize_performance=True, ma
if is_intel_xpu():
return True
if device is None:
device = torch.device("cuda")
props = torch.cuda.get_device_properties(device)
props = torch.cuda.get_device_properties("cuda")
if props.major >= 8:
return True