Skip to content

Commit

Permalink
remove print.
Browse files Browse the repository at this point in the history
  • Loading branch information
sayakpaul committed Nov 19, 2024
1 parent 53bc502 commit f81b71e
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion tests/quantization/bnb/test_4bit.py
Original file line number Diff line number Diff line change
Expand Up @@ -434,7 +434,6 @@ def test_quality(self):
expected_slice = np.array([0.1123, 0.1296, 0.1609, 0.1042, 0.1230, 0.1274, 0.0928, 0.1165, 0.1216])

max_diff = numpy_cosine_similarity_distance(expected_slice, out_slice)
print(f"{max_diff=}")
self.assertTrue(max_diff < 1e-2)

def test_generate_quality_dequantize(self):
Expand Down

0 comments on commit f81b71e

Please sign in to comment.