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", }