Fixing quantization tests (#37650)

* fix

* style

* add capability check
This commit is contained in:
Mohamed Mekkouri
2025-04-22 13:59:57 +02:00
committed by GitHub
parent b3492ff9f7
commit 38c406844e
4 changed files with 18 additions and 4 deletions

View File

@@ -904,6 +904,7 @@ class GgufModelTests(unittest.TestCase):
out = model.generate(text, max_new_tokens=10)
EXPECTED_TEXT = 'Hello with the prompt, "What is the best way'
self.assertEqual(tokenizer.decode(out[0], skip_special_tokens=True), EXPECTED_TEXT)
@require_read_token