wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-12-20 03:23:00 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Fix test.

Browse Source
This commit is contained in:
comfyanonymous 2025-11-30 03:07:03 -05:00 committed by GitHub
parent e1e50e017c
commit b440a2bde9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests-unit/comfy_quant/test_mixed_precision.py
View File

@ -115,7 +115,8 @@ class TestMixedPrecisionOps(unittest.TestCase):
# Forward pass # Forward pass
input_tensor = torch.randn(5, 10, dtype=torch.bfloat16) input_tensor = torch.randn(5, 10, dtype=torch.bfloat16)
output = model(input_tensor) with torch.inference_mode():
output = model(input_tensor)
self.assertEqual(output.shape, (5, 40)) self.assertEqual(output.shape, (5, 40))