Skip to content

Commit e438a38

Browse files
Fix unit test (#495)
* select_quant_linear_with_pack do nothing only call select_quant_linear init QuantLinear * fix test_pt unit test
1 parent 6fd07cd commit e438a38

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_pt.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ def test_pt(self):
1515
)
1616
]
1717

18-
reference_output = "</s>gptqmodel is an easy-to-use model for creating a variety of a variety"
18+
reference_output = "</s>gptqmodel is an easy-to-use model quantization library that can be used"
1919

2020
quantize_config = QuantizeConfig(
2121
bits=4,
@@ -30,6 +30,6 @@ def test_pt(self):
3030

3131
model = GPTQModel.from_quantized(quantized_model_id, device="cuda:0", use_safetensors=False)
3232

33-
result = tokenizer.decode(model.generate(**tokenizer("gptqmodel is an easy-to-use model", return_tensors="pt").to(model.device))[0])
33+
result = tokenizer.decode(model.generate(**tokenizer("gptqmodel is an easy-to-use model quantization library", return_tensors="pt").to(model.device))[0])
3434

3535
self.assertEqual(result, reference_output)

0 commit comments

Comments
 (0)