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

Fix issue.

Browse Source
This commit is contained in:
comfyanonymous 2023-03-03 13:18:01 -05:00
parent 8586515886
commit ebfcf0a9c9
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

@ -185,7 +185,7 @@ def should_use_fp16():
if torch.cuda.is_bf16_supported(): if torch.cuda.is_bf16_supported():
return True return True
props = torch.cuda.get_device_properties() props = torch.cuda.get_device_properties("cuda")
if props.major < 7: if props.major < 7:
return False return False