Update __init__.py

Signed-off-by: mayani-nv <[email protected]>
NVIDIA · Nov 14, 2024 · 622c690 · 622c690
1 parent 4e2d922
commit 622c690
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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",