Make --force-fp32 disable loading models in bf16.

This commit is contained in:
comfyanonymous 2024-02-16 23:01:54 -05:00
parent 97d03ae04a
commit 0b3c50480c

View File

@ -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