We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3e72979 commit a98a184Copy full SHA for a98a184
tests/quantization/torchao/test_torchao.py
@@ -450,8 +450,6 @@ def test_memory_footprint(self):
450
451
# Will quantize all the linear layers except x_embedder
452
for name, module in transformer_int4wo_gs32.named_modules():
453
- if name == "x_embedder":
454
- print(module)
455
if isinstance(module, nn.Linear) and name not in ["x_embedder"]:
456
self.assertTrue(isinstance(module.weight, AffineQuantizedTensor))
457
0 commit comments