Skip to content

Commit

Permalink
make style
Browse files Browse the repository at this point in the history
  • Loading branch information
MekkCyber committed Jan 15, 2025
1 parent 76815d1 commit 32e7e7b
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/quantization/quanto_integration/test_quanto.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@
from transformers.testing_utils import (
require_accelerate,
require_optimum_quanto,
require_read_token,
require_torch_accelerator,
require_torch_gpu,
slow,
Expand Down Expand Up @@ -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 = [
Expand Down

0 comments on commit 32e7e7b

Please sign in to comment.