Fix CI rules

Luca-Calabria · Dec 6, 2024 · 1158a5c · 1158a5c
1 parent ed61be8
commit 1158a5c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -326,6 +326,8 @@ def to_test(
             return True
         elif "huggyllama" in model_name and IS_GAUDI2 and deepspeed:
             return True
+        elif "gemma" in model_name and IS_GAUDI2:
+            return True
 
         return False