From d4a54ff96cc54ae5e410b6c3dca63fa654a4375b Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Sat, 10 Jan 2026 01:32:28 -0500
Subject: [PATCH] Fix test.

---
 tests-unit/comfy_quant/test_mixed_precision.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests-unit/comfy_quant/test_mixed_precision.py b/tests-unit/comfy_quant/test_mixed_precision.py
index 4922fbcdc..f02231d44 100644
--- a/tests-unit/comfy_quant/test_mixed_precision.py
+++ b/tests-unit/comfy_quant/test_mixed_precision.py
@@ -153,7 +153,7 @@ class TestMixedPrecisionOps(unittest.TestCase):
         state_dict2 = model.state_dict()
 
         # Verify layer1.weight is a QuantizedTensor with scale preserved
-        self.assertEqual(state_dict2["layer1.weight"], fp8_weight)
+        self.assertTrue(torch.equal(state_dict2["layer1.weight"], fp8_weight))
         self.assertEqual(state_dict2["layer1.weight_scale"].item(), 3.0)
         self.assertEqual(model.layer1.weight._layout_cls, "TensorCoreFP8E4M3Layout")