mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-01-11 02:15:17 +00:00
Make --force-fp32 disable loading models in bf16.
This commit is contained in:
parent
97d03ae04a
commit
0b3c50480c
@ -772,6 +772,9 @@ def should_use_fp16(device=None, model_params=0, prioritize_performance=True, ma
|
||||
return True
|
||||
|
||||
def should_use_bf16(device=None):
|
||||
if FORCE_FP32:
|
||||
return False
|
||||
|
||||
if is_intel_xpu():
|
||||
return True
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user