diff --git a/orttraining/orttraining/python/training/ortmodule/_onnx_models.py b/orttraining/orttraining/python/training/ortmodule/_onnx_models.py
index 9cd88f56d7..55c1970613 100644
--- a/orttraining/orttraining/python/training/ortmodule/_onnx_models.py
+++ b/orttraining/orttraining/python/training/ortmodule/_onnx_models.py
@@ -3,7 +3,6 @@
 # _onnx_models.py
 
 from dataclasses import dataclass
-from filelock import SoftFileLock
 import onnx
 import os
 import torch