From 9a15e22f5e9416c0968ce3de33923f8f601257dd Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sat, 1 Mar 2025 12:41:50 +0200
Subject: [PATCH] Update model_optimization_nodes.py

---
 nodes/model_optimization_nodes.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/nodes/model_optimization_nodes.py b/nodes/model_optimization_nodes.py
index 54eec62..5f6f3df 100644
--- a/nodes/model_optimization_nodes.py
+++ b/nodes/model_optimization_nodes.py
@@ -210,6 +210,7 @@ class DiffusionModelLoaderKJ(BaseLoaderKJ):
         model = comfy.sd.load_diffusion_model(unet_path, model_options=model_options)
         if dtype := DTYPE_MAP.get(compute_dtype):
             model.set_model_compute_dtype(dtype)
+            model.force_cast_weights = False
         self._patch_modules(patch_cublaslinear, sage_attention)
         
         return (model,)