Add Phi-2 to model presets (ludwig-ai#3912)

vijayi1 · Jan 23, 2024 · 92389a8 · 92389a8
1 parent 5c0221c
commit 92389a8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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",
 }