From 0ed23e4db26b0de01765c7d032002fe36702d357 Mon Sep 17 00:00:00 2001 From: Marc Sun <57196510+SunMarc@users.noreply.github.com> Date: Wed, 16 Aug 2023 11:02:57 -0400 Subject: [PATCH] fix vit hybrid test (#25543) fix test --- tests/models/vit_hybrid/test_modeling_vit_hybrid.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/vit_hybrid/test_modeling_vit_hybrid.py b/tests/models/vit_hybrid/test_modeling_vit_hybrid.py index a1146264894..3ea407eafd4 100644 --- a/tests/models/vit_hybrid/test_modeling_vit_hybrid.py +++ b/tests/models/vit_hybrid/test_modeling_vit_hybrid.py @@ -281,7 +281,7 @@ class ViTModelIntegrationTest(unittest.TestCase): image = prepare_img() - inputs = image_processor(images=image, return_tensors="pt") + inputs = image_processor(images=image, return_tensors="pt").to(torch_device) outputs = model(**inputs) logits = outputs.logits # model predicts one of the 1000 ImageNet classes