Tag tests as slow (#21537)

begone slow tests
This commit is contained in:
Joao Gante
2023-02-09 14:46:15 +00:00
committed by GitHub
parent 3a726777ca
commit 0d33381fad
5 changed files with 13 additions and 0 deletions

View File

@@ -3305,6 +3305,7 @@ class ModelUtilsTest(TestCasePlus):
_ = ModelWithHead.from_pretrained(tmp_dir)
@require_torch_gpu
@slow
def test_pretrained_low_mem_new_config(self):
# Checking for 1 model(the same one which was described in the issue) .
model_ids = ["gpt2"]