Update README.md

README.md

@@ -37,7 +37,7 @@ For the benchmark, 0-shot and default prompts are used.
 ## Usage
 Install the required libraries as follows:
 ```sh
->>> python -m pip install numpy sentencepiece torch transformers accelerate transformers_stream_generator tiktoken
+>>> python -m pip install numpy sentencepiece torch transformers accelerate transformers_stream_generator tiktoken einops
 ```
 
 Execute the following python code: