diff --git a/requirements.txt b/requirements.txt
index 369dbd4..9b02e26 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -27,6 +27,7 @@ https://github.com/woct0rdho/triton-windows/releases/download/v3.1.0-windows.pos
 transformers>=4.49.0
 accelerate
 sentence-transformers
+datasets
 
 # Qwen model dependencies
 tokenizers>=0.15.0