From 92389a842a1aab5e986faec62e13fe6f12c4a908 Mon Sep 17 00:00:00 2001
From: Arnav Garg <106701836+arnavgarg1@users.noreply.github.com>
Date: Tue, 23 Jan 2024 12:14:14 -0500
Subject: [PATCH] Add Phi-2 to model presets (#3912)

---
 ludwig/schema/llms/base_model.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ludwig/schema/llms/base_model.py b/ludwig/schema/llms/base_model.py
index 59cf77ff3e7..fe77291bfc5 100644
--- a/ludwig/schema/llms/base_model.py
+++ b/ludwig/schema/llms/base_model.py
@@ -56,6 +56,10 @@
     # Zephyr
     "zephyr-7b-alpha": "HuggingFaceH4/zephyr-7b-alpha",
     "zephyr-7b-beta": "HuggingFaceH4/zephyr-7b-beta",
+    # Phi
+    "phi-1": "microsoft/phi-1",
+    "phi-1_5": "microsoft/phi-1_5",
+    "phi-2": "microsoft/phi-2",
 }