jinv2 commited on
Commit
35c9492
·
verified ·
1 Parent(s): f20d838

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -2
requirements.txt CHANGED
@@ -4,5 +4,6 @@ peft>=0.9.0
4
  torch>=2.0.0
5
  accelerate>=0.28.0
6
  sentencepiece
7
- tiktoken # <--- 新增这一行
8
- # transformers_stream_generator # Qwen也可能需要这个,如果tiktoken之后还报错,可以加上这个
 
 
4
  torch>=2.0.0
5
  accelerate>=0.28.0
6
  sentencepiece
7
+ tiktoken
8
+ einops # <--- 新增
9
+ transformers_stream_generator # <--- 新增 (或确保它在)