diff --git a/AITrain/requirements.txt b/AITrain/requirements.txt
index 2a0f3ec..cdba7ef 100644
--- a/AITrain/requirements.txt
+++ b/AITrain/requirements.txt
@@ -7,4 +7,11 @@ transformers>=4.21.0
 datasets
 scikit-learn
 wandb
-accelerate
\ No newline at end of file
+accelerate
+modelscope
+vllm
+swanlab==0.5.7
+accelerate==1.6.0
+datasets==3.5.1
+peft==0.15.2
+autoawq