From 0b3c50480c8cfcfe22c2a4059f91cec337114a78 Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Fri, 16 Feb 2024 23:01:54 -0500
Subject: [PATCH] Make --force-fp32 disable loading models in bf16.

---
 comfy/model_management.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index eb7178b4..f0f4ebf5 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -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