更新安装依赖包

2025-08-09 08:23:51 +08:00 · 2025-08-09 08:23:51 +08:00 · 731ad3d099
commit 731ad3d099
parent 150ec134b2
1 changed files with 8 additions and 1 deletions
--- a/AITrain/requirements.txt
+++ b/AITrain/requirements.txt
@ -7,4 +7,11 @@ transformers>=4.21.0
 datasets
 scikit-learn
 wandb
-accelerate
+accelerate
+modelscope
+vllm
+swanlab==0.5.7
+accelerate==1.6.0
+datasets==3.5.1
+peft==0.15.2
+autoawq