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

Speedup on some models by not upcasting bfloat16 to float32 on mac.

Browse Source
This commit is contained in:
comfyanonymous
2025-02-24 05:41:07 -05:00
parent 4553891bbd
commit 96d891cb94
2 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/model_management.py
View File

@@ -954,7 +954,7 @@ def force_upcast_attention_dtype():
upcast = True
if upcast:
return torch.float32
return {torch.float16: torch.float32}
else:
return None