Browse Source

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

pull/2811/head
comfyanonymous 9 months ago
parent
commit
0b3c50480c
  1. 3
      comfy/model_management.py

3
comfy/model_management.py

@ -772,6 +772,9 @@ def should_use_fp16(device=None, model_params=0, prioritize_performance=True, ma
return True return True
def should_use_bf16(device=None): def should_use_bf16(device=None):
if FORCE_FP32:
return False
if is_intel_xpu(): if is_intel_xpu():
return True return True

Loading…
Cancel
Save