diff --git a/docs/chapter6/code/requirements.txt b/docs/chapter6/code/requirements.txt
index c006a46..4e3499a 100644
--- a/docs/chapter6/code/requirements.txt
+++ b/docs/chapter6/code/requirements.txt
@@ -1,7 +1,7 @@
 transformers
 datasets
 torch
-torchdata
+torchdata==0.9.0
 deepspeed
 pandas
-swanlab
\ No newline at end of file
+swanlab