From 58850b7e1b25a737568b967986486e146df0158a Mon Sep 17 00:00:00 2001 From: Shuai Zhang Date: Wed, 18 Sep 2024 10:47:44 +0800 Subject: [PATCH] Update __init__.py --- msamp/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/msamp/__init__.py b/msamp/__init__.py index c0841e1c..f43a05dc 100644 --- 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