File tree Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -14,6 +14,6 @@ python3 -c "import torch; import torchao; print(f'Torch version: {torch.__versio
1414
1515pip install pytest expecttest parameterized accelerate hf_transfer ' modelscope!=1.15.0'
1616
17- # pytest -v -s torchao/test/quantization/quantize_/workflows/int4/test_int4_plain_int32_tensor.py
17+ pytest -v -s torchao/test/quantization/quantize_/workflows/int4/test_int4_plain_int32_tensor.py
1818
19- pytest -v -s torchao/test/quantization/test_observer.py::TestQuantFlow::test_fixed_qparams_observer
19+ pytest -v -s torchao/test/quantization/
Original file line number Diff line number Diff line change @@ -187,7 +187,6 @@ def test_fixed_qparams_observer(self):
187187 obs (example_input )
188188 obs .set_qparams (torch .ones (2048 ))
189189 scale , zero_point = obs .calculate_qparams ()
190- print ("scale" , scale )
191190 self .assertTrue (torch .allclose (scale , torch .ones (2048 )))
192191
193192
You can’t perform that action at this time.
0 commit comments