diff --git a/tests/models/test_modeling_common.py b/tests/models/test_modeling_common.py
index 7794472f9a85..57431d8b161b 100644
--- a/tests/models/test_modeling_common.py
+++ b/tests/models/test_modeling_common.py
@@ -71,6 +71,7 @@
     require_torch_2,
     require_torch_accelerator,
     require_torch_accelerator_with_training,
+    require_torch_gpu,
     require_torch_multi_accelerator,
     run_test_in_subprocess,
     slow,