saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

make style

Browse source
This commit is contained in:
MekkCyber 2025-01-15 17:15:49 +00:00
parent 76815d1360
commit 32e7e7b6b1
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/quantization/quanto_integration/test_quanto.py
View file

@ -20,7 +20,6 @@ from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, Quanto
from transformers.testing_utils import (
require_accelerate,
require_optimum_quanto,
require_read_token,
require_torch_accelerator,
require_torch_gpu,
slow,
@ -448,7 +447,8 @@ class QuantoKVCacheQuantizationTest(unittest.TestCase):
@slow
def test_quantized_cache(self):
EXPECTED_TEXT_COMPLETION = [
'Simply put, the theory of relativity states that ', 'My favorite all time favorite condiment is ketchup. I love it because it is so easy to make and it is so good for you. I love ketchup because it is so good for you. I love ketchup because it is so'
"Simply put, the theory of relativity states that ",
"My favorite all time favorite condiment is ketchup. I love it because it is so easy to make and it is so good for you. I love ketchup because it is so good for you. I love ketchup because it is so",
]
prompts = [