diff --git a/examples/training/llama/requirements.txt b/examples/training/llama/requirements.txt
index d05b079..208ef03 100644
--- a/examples/training/llama/requirements.txt
+++ b/examples/training/llama/requirements.txt
@@ -1,5 +1,5 @@
 transformers==4.31.0
 regex
 tensorboard
-datasets
+datasets==3.0.2
 sentencepiece