Update __init__.py

Azure · Sep 18, 2024 · 58850b7 · 58850b7
1 parent 8de09e0
commit 58850b7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/msamp/__init__.py b/msamp/__init__.py
@@ -1,7 +1,10 @@
 # Copyright (c) Microsoft Corporation.
 # Licensed under the MIT License.
 
-"""MS-AMP Python module."""
+"""MS-AMP Python module.
+
+This module is top level module.
+"""
 
 import torch
 from deepspeed.ops.adam import FusedAdam