From 622c690db85689c44aa4d4d555a1740ea3f8dce5 Mon Sep 17 00:00:00 2001
From: mayani-nv <67936769+mayani-nv@users.noreply.github.com>
Date: Thu, 14 Nov 2024 12:09:56 -0800
Subject: [PATCH] Update __init__.py

Signed-off-by: mayani-nv <67936769+mayani-nv@users.noreply.github.com>
---
 nemo/collections/llm/gpt/model/__init__.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/nemo/collections/llm/gpt/model/__init__.py b/nemo/collections/llm/gpt/model/__init__.py
index b42ceac564bc..859aa9ed5f6e 100644
--- a/nemo/collections/llm/gpt/model/__init__.py
+++ b/nemo/collections/llm/gpt/model/__init__.py
@@ -79,6 +79,11 @@
     NemotronConfig,
     NemotronModel,
 )
+from nemo.collections.llm.gpt.model.phi3 import (
+    Phi3Config,
+    Phi3ConfigMini,
+    Phi3Model,
+)
 from nemo.collections.llm.gpt.model.qwen2 import (
     Qwen2Config,
     Qwen2Config1P5B,
@@ -140,6 +145,9 @@
     "Nemotron3Config22B",
     "Nemotron4Config340B",
     "NemotronModel",
+    "Phi3Config",
+    "Phi3ConfigMini",
+    "Phi3Model",
     "CodeLlamaConfig7B",
     "CodeLlamaConfig13B",
     "CodeLlamaConfig34B",