Update requirements.txt
Browse files- requirements.txt +3 -0
requirements.txt
CHANGED
@@ -2,3 +2,6 @@ accelerate==0.33.0
|
|
2 |
bitsandbytes==0.43.3
|
3 |
transformers==4.44.0
|
4 |
tiktoken==0.3.0
|
|
|
|
|
|
|
|
2 |
bitsandbytes==0.43.3
|
3 |
transformers==4.44.0
|
4 |
tiktoken==0.3.0
|
5 |
+
einops
|
6 |
+
transformers_stream_generator
|
7 |
+
|