remove unused comments

huggingface · Oct 24, 2024 · bd11280 · bd11280
1 parent 06011a4
commit bd11280
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/tests/quantization/ggml/test_ggml.py b/tests/quantization/ggml/test_ggml.py
@@ -738,7 +738,6 @@ def test_starcoder2_weights_conversion_fp16(self):
 
         for layer_name, original_params in original_state_dict.items():
             if layer_name in converted_state_dict:
-                # quantized models do not contain "lm_head.weight" layer
                 self.assertTrue(original_params.shape == converted_state_dict[layer_name].shape)
                 torch.testing.assert_close(original_params, converted_state_dict[layer_name])
             else:
@@ -777,7 +776,6 @@ def test_mamba_weights_conversion_fp16(self):
 
         for layer_name, original_params in original_state_dict.items():
             if layer_name in converted_state_dict:
-                # quantized models do not contain "lm_head.weight" layer
                 self.assertTrue(original_params.shape == converted_state_dict[layer_name].shape)
                 if "mixer.A_log" in layer_name:
                     # we should increase tolerance after exponential reversing