Skip to content

GitLab

Unverified Commit 74e84f1f authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

make blenderbot test slow (#11395)

parent c3d6f339
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_modeling_blenderbot_small.py
View file @ 74e84f1f
...@@ -305,6 +305,7 @@ class Blenderbot90MIntegrationTests(unittest.TestCase): ...@@ -305,6 +305,7 @@ class Blenderbot90MIntegrationTests(unittest.TestCase):
"i'm not sure. i just feel like i've been feeling like i have to be in a certain place", "i'm not sure. i just feel like i've been feeling like i have to be in a certain place",
) )
@slow
def test_90_generation_from_short_input(self): def test_90_generation_from_short_input(self):
model_inputs = self.tokenizer(["sam"], return_tensors="pt").to(torch_device) model_inputs = self.tokenizer(["sam"], return_tensors="pt").to(torch_device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment