r1208 commited on
Commit
2e41698
·
verified ·
1 Parent(s): b86b1f9

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -2
requirements.txt CHANGED
@@ -1,8 +1,7 @@
1
- torch==2.1.2+cu118
2
  transformers
3
  peft==0.11.1
4
  accelerate==0.30.1
5
  trl==0.9.4
6
  flash_attn==2.5.4
7
  bitsandbytes==0.43.1
8
- datasets==2.19.1
 
1
+ torch==2.1.2
2
  transformers
3
  peft==0.11.1
4
  accelerate==0.30.1
5
  trl==0.9.4
6
  flash_attn==2.5.4
7
  bitsandbytes==0.43.1