Skip to content

GitLab

Commit 0b3c5048 authored by comfyanonymous's avatar comfyanonymous
Browse files

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

parent 97d03ae0
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
comfy/model_management.py
View file @ 0b3c5048
...@@ -772,6 +772,9 @@ def should_use_fp16(device=None, model_params=0, prioritize_performance=True, ma ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment