Llama in modelling value head tests (#1878)

huggingface · Jul 26, 2024 · 82b07d6 · 82b07d6
1 parent 72bf6c2
commit 82b07d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_modeling_value_head.py b/tests/test_modeling_value_head.py
@@ -33,7 +33,7 @@
     "trl-internal-testing/tiny-random-CodeGenForCausalLM-sharded",
     "trl-internal-testing/tiny-random-GPTNeoXForCausalLM-safetensors-sharded",
     "trl-internal-testing/tiny-random-GPTNeoXForCausalLM-safetensors",
-    # "trl-internal-testing/tiny-random-LlamaForCausalLM", uncomment on the next transformers release
+    "trl-internal-testing/tiny-random-LlamaForCausalLM",
 ]
 
 ALL_SEQ2SEQ_MODELS = [