From 4ee951a3575a459d41766469bf642e82982f95f8 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Wed, 25 Mar 2026 23:17:28 -0400
Subject: [PATCH] Update.

---
 comfy/ops.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/comfy/ops.py b/comfy/ops.py
index ca25693db..b5cd1d47e 100644
--- a/comfy/ops.py
+++ b/comfy/ops.py
@@ -928,6 +928,7 @@ def mixed_precision_ops(quant_config={}, compute_dtype=torch.bfloat16, full_prec
                 weight = state_dict.pop(weight_key, None)
                 if weight is None:
                     logging.warning(f"Missing weight for layer {layer_name}")
+                    self.weight = None
                     return
 
                 manually_loaded_keys = [weight_key]
@@ -1034,6 +1035,9 @@ def mixed_precision_ops(quant_config={}, compute_dtype=torch.bfloat16, full_prec
                 if self.bias is not None:
                     sd["{}bias".format(prefix)] = self.bias
 
+                if self.weight is None:
+                    return sd
+
                 if isinstance(self.weight, QuantizedTensor):
                     sd_out = self.weight.state_dict("{}weight".format(prefix))
                     for k in sd_out: