diff --git a/fastNLP/core/utils/seq_len_to_mask.py b/fastNLP/core/utils/seq_len_to_mask.py
index 5458602d..c3d0f9ec 100644
--- a/fastNLP/core/utils/seq_len_to_mask.py
+++ b/fastNLP/core/utils/seq_len_to_mask.py
@@ -94,4 +94,4 @@ def seq_len_to_mask(seq_len, max_len: Optional[int]=None):
         pass
 
     raise TypeError("seq_len_to_mask function only supports numpy.ndarray, torch.Tensor, paddle.Tensor, "
-                    f"and jittor.Var, but got {type(seq_len)}")
\ No newline at end of file
+                    f"jittor.Var and oneflow.Tensor, but got {type(seq_len)}")
\ No newline at end of file