diff --git a/requirements.txt b/requirements.txt
index bf83b003..2e389577 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ numpy
 accelerate>=0.20.3
 transformers
 torch
-aim==3.17.5
+aim==3.18.1
 sentencepiece
 tokenizers>=0.13.3
 tqdm