From e43a73e003a6bded93d1722efec49242b187cc76 Mon Sep 17 00:00:00 2001 From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com> Date: Sat, 10 Jan 2026 01:59:52 -0500 Subject: [PATCH] Fix. --- 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 f02231d44..7c740491d 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.assertTrue(torch.equal(state_dict2["layer1.weight"], fp8_weight)) + self.assertTrue(torch.equal(state_dict2["layer1.weight"].view(torch.uint8), fp8_weight.view(torch.uint8))) self.assertEqual(state_dict2["layer1.weight_scale"].item(), 3.0) self.assertEqual(model.layer1.weight._layout_cls, "TensorCoreFP8E4M3Layout")