From dfebfb2a965876780e0ee1ce313ec13f8e5cc94f Mon Sep 17 00:00:00 2001
From: sayakpaul <spsayakpaul@gmail.com>
Date: Mon, 28 Apr 2025 13:24:48 +0800
Subject: [PATCH] fix import.

---
 tests/models/test_modeling_common.py | 1 +
 1 file changed, 1 insertion(+)

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,