Add deprecation notice to modules (#15050)

* Add deprecation notices for collections and lightning

Signed-off-by: Charlie Truong <[email protected]>

* Fix warning

Signed-off-by: Charlie Truong <[email protected]>

* Fix diffusion deprecation notice

Signed-off-by: Charlie Truong <[email protected]>

* Update deprecation notice

Signed-off-by: Charlie Truong <[email protected]>

* Fix nemo2 deprecation notice

Signed-off-by: Charlie Truong <[email protected]>

* Update deprecation notice for vlm

Signed-off-by: Charlie Truong <[email protected]>

* Fix multimodal_autoregressive module notice

Signed-off-by: Charlie Truong <[email protected]>

---------

Signed-off-by: Charlie Truong <[email protected]>

Author: chtruong814

nemo/collections/avlm/__init__.py

@@ -25,3 +25,13 @@
     "AVLMModel",
     "avlm_8b",
 ]
+
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.avlm is deprecated and will be removed in a future major NeMo FW container release. "
+    "Please refer to the new Megatron-Bridge repository: https://github.com/NVIDIA-NeMo/Megatron-Bridge",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/diffusion/__init__.py

@@ -11,3 +11,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+import warnings
+
+
+warnings.warn(
+    "nemo.collections.diffusion is deprecated and will be removed in a future major NeMo FW container release.",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/llm/__init__.py

@@ -447,3 +447,14 @@
     __all__.append("evaluate")
 except ImportError as error:
     logging.warning(f"The evaluate module could not be imported: {error}")
+
+
+import warnings
+
+
+warnings.warn(
+    "nemo.collections.llm is deprecated and will be removed in a future major NeMo FW container release. "
+    "Please refer to the new Megatron-Bridge repository: https://github.com/NVIDIA-NeMo/Megatron-Bridge",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/multimodal/__init__.py

@@ -11,3 +11,12 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.multimodal is deprecated and will be removed in a future major NeMo FW container release. "
+    "Please refer to the new Megatron-Bridge repository: https://github.com/NVIDIA-NeMo/Megatron-Bridge",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/multimodal_autoregressive/__init__.py

@@ -0,0 +1,23 @@
+# Copyright (c) 2025, NVIDIA CORPORATION.  All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.multimodal_autoregressive is deprecated and will be removed in a future major NeMo "
+    "FW container release. Please refer to the new Megatron-Bridge repository: "
+    "https://github.com/NVIDIA-NeMo/Megatron-Bridge",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/nlp/__init__.py

@@ -25,3 +25,12 @@
 
 # Set collection name.
 __description__ = "Natural Language Processing collection"
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.nlp is deprecated and will be removed in a future major NeMo FW container release."
+    "Please refer to the new Megatron-Bridge repository: https://github.com/NVIDIA-NeMo/Megatron-Bridge",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/vision/__init__.py

@@ -23,3 +23,12 @@
 
 # Set collection name.
 __description__ = "Computer Vision collection"
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.nlp is deprecated and will be removed in a future major NeMo FW container release. "
+    "Please refer to the new Megatron-Bridge repository: https://github.com/NVIDIA-NeMo/Megatron-Bridge",
+    DeprecationWarning,
+    stacklevel=2,
+)

nemo/collections/vlm/__init__.py

@@ -184,3 +184,12 @@
     from nemo.utils import logging
 
     logging.warning(f"Failed to import nemo.collections.vlm.api: {error}")
+
+import warnings
+
+warnings.warn(
+    "nemo.collections.vlm is deprecated and will be removed in a future major NeMo FW container release. "
+    "Please refer to the new Megatron-Bridge repository: https://github.com/NVIDIA-NeMo/Megatron-Bridge",
+    DeprecationWarning,
+    stacklevel=2,
+)