[tests] fix import. (#11434)

sayakpaul · web-flow · commit 0e3f2713c2c0 · 2025-04-28T13:32:28.000+08:00
fix import.
diff --git 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,
