diff --git a/pyproject.toml b/pyproject.toml
index 56b44226..b2528fb8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -56,7 +56,7 @@ dependencies = [
     "diffusers==0.32.0", # Diffusion models published on Hugging Face
     "timm==1.0.12", # Image models
     "datasets==3.2.0", # Datasets published on Hugging Face
-    "x-transformers==1.43.2", # Transformer utils
+    "x-transformers==1.43.4", # Transformer utils
     "mambapy==1.2.0", # Mamba
     "denoising-diffusion-pytorch==2.1.1", # Diffusion models
     "vector-quantize-pytorch==1.20.11", # Vector quantization utils