From 74e84f1fa677e0276be700553544a52583f10867 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Fri, 23 Apr 2021 13:49:09 +0200 Subject: [PATCH] make blenderbot test slow (#11395) --- tests/test_modeling_blenderbot_small.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_modeling_blenderbot_small.py b/tests/test_modeling_blenderbot_small.py index a4c68085f6..0eea63c8c6 100644 --- a/tests/test_modeling_blenderbot_small.py +++ b/tests/test_modeling_blenderbot_small.py @@ -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", ) + @slow def test_90_generation_from_short_input(self): model_inputs = self.tokenizer(["sam"], return_tensors="pt").to(torch_device)