diff --git a/cosyvoice/utils/train_utils.py b/cosyvoice/utils/train_utils.py
index 8678422..eab92f8 100644
--- a/cosyvoice/utils/train_utils.py
+++ b/cosyvoice/utils/train_utils.py
@@ -14,7 +14,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from contextlib import nullcontext
 import logging
 import os
 import torch