From b844f8a9ab5bbbbf8001e2d83c75e85eaf40549c Mon Sep 17 00:00:00 2001
From: Younes Belkada <49240599+younesbelkada@users.noreply.github.com>
Date: Wed, 29 Mar 2023 17:40:45 +0200
Subject: [PATCH] [`Pix2Struct`] Fix slow test (#22448)

fix slow test
---
 tests/models/pix2struct/test_modeling_pix2struct.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/models/pix2struct/test_modeling_pix2struct.py b/tests/models/pix2struct/test_modeling_pix2struct.py
index dc219bbd61a..f56f8f6d3ec 100644
--- a/tests/models/pix2struct/test_modeling_pix2struct.py
+++ b/tests/models/pix2struct/test_modeling_pix2struct.py
@@ -406,7 +406,7 @@ class Pix2StructTextImageModelTest(ModelTesterMixin, unittest.TestCase):
     def test_model(self):
         config, input_dict = self.model_tester.prepare_config_and_inputs_for_common()
         for model_class in self.all_model_classes:
-            model = model_class(config)
+            model = model_class(config).to(torch_device)
 
             output = model(**input_dict)
             self.assertEqual(