diff --git a/requirements.txt b/requirements.txt
index c247929..0f4aca9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,6 +2,6 @@ einops
 omegaconf>=2.3.0
 diffusers>=v0.33.1
 pytorch-extension
-rotary_embedding_torch
+rotary_embedding_torch>=0.5.3
 opencv-python
 peft>=0.15.0