fix

add Gemma2 for causal LM to test classes
2025-10-21 01:23:56 +08:00 · 2024-08-09 11:20:38 +02:00 · 2024-08-09 11:13:17 +02:00
1 changed files with 1 additions and 1 deletions
--- a/tests/models/gemma2/test_modeling_gemma2.py
+++ b/tests/models/gemma2/test_modeling_gemma2.py
@ -59,7 +59,7 @@ class Gemma2ModelTest(GemmaModelTest, unittest.TestCase):
        if is_torch_available()
        else ()
    )
-    all_generative_model_classes = ()
+    all_generative_model_classes = (Gemma2ForCausalLM,) if is_torch_available() else ()
    pipeline_model_mapping = (
        {
            "feature-extraction": Gemma2Model,
Author	SHA1	Message	Date
Pablo Montalvo	5b64dfe718	fix	2024-08-09 11:20:38 +02:00
Pablo Montalvo	90b20fc1db	add Gemma2 for causal LM to test classes	2024-08-09 11:13:17 +02:00