WinterGYC commited on
Commit
972ba8e
1 Parent(s): 38f2963

Update requirements.

Browse files
Files changed (1) hide show
  1. requirements.txt +7 -3
requirements.txt CHANGED
@@ -1,5 +1,9 @@
1
- accelerate
2
- colorama
3
  cpm_kernels
 
 
 
4
  sentencepiece
5
- transformers_stream_generator
 
 
1
+ protobuf
2
+ transformers==4.30.2
3
  cpm_kernels
4
+ torch>=2.0
5
+ # gradio
6
+ mdtex2html
7
  sentencepiece
8
+ accelerate
9
+ loguru