OrionZheng commited on
Commit
3ccc470
1 Parent(s): a22eba0

Update colab_env.txt

Browse files
Files changed (1) hide show
  1. colab_env.txt +1 -1
colab_env.txt CHANGED
@@ -16,7 +16,7 @@ tensorflow-cpu==2.15.0.post1
16
  tensorstore==0.1.45
17
  protobuf==3.20.3
18
  colossalai >= 0.3.3
19
- torch >= 1.8.1
20
  transformers==4.34.0
21
  sentencepiece==0.1.99
22
  datasets==2.14.7
 
16
  tensorstore==0.1.45
17
  protobuf==3.20.3
18
  colossalai >= 0.3.3
19
+ torch >= 2.1.0
20
  transformers==4.34.0
21
  sentencepiece==0.1.99
22
  datasets==2.14.7