add Phi

huggingface · Jan 15, 2024 · 350529e · 350529e
1 parent 573d8da
commit 350529e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/slow/testing_constants.py b/tests/slow/testing_constants.py
@@ -13,7 +13,11 @@
 # limitations under the License.
 
 # TODO: push them under trl-org
-MODELS_TO_TEST = ["HuggingFaceM4/tiny-random-LlamaForCausalLM", "HuggingFaceM4/tiny-random-MistralForCausalLM"]
+MODELS_TO_TEST = [
+    "HuggingFaceM4/tiny-random-LlamaForCausalLM",
+    "HuggingFaceM4/tiny-random-MistralForCausalLM",
+    "hf-internal-testing/tiny-random-PhiForCausalLM",
+]
 
 # We could have also not declared these variables but let's be verbose
 PACKING_OPTIONS = [True, False]