diff --git a/requirements.txt b/requirements.txt
index 069c30b..8757347 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -61,6 +61,6 @@ torchmetrics==1.6.0
 
 timm
 pythreejs
-torchdiffe
+torchdiffeq
 deepspeed