Skipp large gemma2 test on Linux CI (#84)

Co-authored-by: Guang Yang <[email protected]>

Author: guangy10

tests/models/test_modeling_gemma2.py

@@ -17,6 +17,7 @@
 import logging
 import os
 import subprocess
+import sys
 import tempfile
 import unittest
 
@@ -32,10 +33,14 @@
 from ..utils import check_causal_lm_output_quality
 
 
+is_linux_ci = sys.platform.startswith("linux") and os.environ.get("GITHUB_ACTIONS") == "true"
+
+
 @pytest.mark.skipif(
     parse(torchao.__version__) < parse("0.11.0.dev0"),
     reason="Only available on torchao >= 0.11.0.dev0",
 )
+@pytest.mark.skipif(is_linux_ci, reason="OOM on linux runner")
 class ExecuTorchModelIntegrationTest(unittest.TestCase):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)