diff --git a/train/requirements.txt b/train/requirements.txt
index dbc4f76..02cba9f 100644
--- a/train/requirements.txt
+++ b/train/requirements.txt
@@ -9,5 +9,5 @@ tensorflow < 2.16 # for tensorflow-addons, so only needed in training
 tf-keras < 2.16 # avoid keras 3 (also needs TF_USE_LEGACY_KERAS=1)
 tf-data < 2.16 # for tensorflow-addons, so only needed in training
 protobuf < 5 # for tensorflow-addons, so only needed in training
-tf2onnx
+eynollah-fork-tf2onnx == 1.17.0.post1
 ml_dtypes >= 0.5