MrYXJ commited on
Commit
d927324
1 Parent(s): bb18cfc

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -1
requirements.txt CHANGED
@@ -1,8 +1,9 @@
1
  accelerate @ git+https://github.com/huggingface/accelerate
2
  bitsandbytes==0.41.1
3
  sentencepiece
4
- calflops==0.2.8
5
  einops
 
6
  huggingface_hub
7
  pandas
8
  transformers
 
1
  accelerate @ git+https://github.com/huggingface/accelerate
2
  bitsandbytes==0.41.1
3
  sentencepiece
4
+ calflops>=0.3.1
5
  einops
6
+ gpt-neox
7
  huggingface_hub
8
  pandas
9
  transformers