SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

make blenderbot test slow (#11395)

Browse Source
This commit is contained in:
Patrick von Platen
2021-04-23 13:49:09 +02:00
committed by GitHub
parent c3d6f33918
commit 74e84f1fa6
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_modeling_blenderbot_small.py
View File

@@ -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)