diff --git a/examples/opensora_hpcai/opensora/models/vae/vae.py b/examples/opensora_hpcai/opensora/models/vae/vae.py
index 1c32818da1..d846d2fdac 100644
--- a/examples/opensora_hpcai/opensora/models/vae/vae.py
+++ b/examples/opensora_hpcai/opensora/models/vae/vae.py
@@ -4,7 +4,7 @@
 from transformers import PretrainedConfig
 
 import mindspore as ms
-from mindspore import nn, ops
+from mindspore import mint, nn, ops
 from mindspore.communication import get_group_size
 
 from ...acceleration.communications import GatherFowardSplitBackward, SplitFowardGatherBackward