use bigger machine for slow tests

huggingface · Jan 28, 2025 · 41abf7f · 41abf7f
1 parent 1827450
commit 41abf7f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.github/workflows/test_onnxruntime_slow.yml b/.github/workflows/test_onnxruntime_slow.yml
@@ -27,7 +27,8 @@ jobs:
       contains(github.event.pull_request.labels.*.name, 'onnxruntime-slow')
       }}
 
-    runs-on: ubuntu-20.04
+    runs-on:
+      group: aws-general-8-plus
 
     steps:
       - name: Checkout

diff --git a/tests/onnxruntime/test_modeling.py b/tests/onnxruntime/test_modeling.py
@@ -1042,7 +1042,7 @@ def test_save_load_ort_model_with_external_data(self):
     def test_save_load_decoder_model_with_external_data(self, use_cache: bool):
         with tempfile.TemporaryDirectory() as tmpdirname:
             model = ORTModelForCausalLM.from_pretrained(
-                "gpt2-large", use_cache=use_cache, export=True, use_merged=False, use_io_binding=False
+                "gpt2-large", export=True, use_cache=use_cache, use_merged=False, use_io_binding=False
             )
             model.save_pretrained(tmpdirname)