[CI] revert device in test_export_static_cache (#39662)

* revert device * add todo
2025-07-25 16:36:12 +01:00
parent 850bdeaa95
commit 5d0ba3e479
10 changed files with 10 additions and 10 deletions
--- a/tests/models/phi3/test_modeling_phi3.py
+++ b/tests/models/phi3/test_modeling_phi3.py
@@ -384,7 +384,7 @@ class Phi3IntegrationTest(unittest.TestCase):
            config.rope_scaling["type"] = "default"

        # Load model
-        device = torch_device
+        device = "cpu"  # TODO (joao / export experts): should be on `torch_device`, but causes GPU OOM
        dtype = torch.bfloat16
        cache_implementation = "static"
        attn_implementation = "sdpa"