Update README.md
Browse files
README.md
CHANGED
|
@@ -36,7 +36,7 @@ Or you can use the tokenizer's chat_template, as shown below.
|
|
| 36 |
|
| 37 |
Dependencies
|
| 38 |
```
|
| 39 |
-
pip install -u torch[cuda]
|
| 40 |
```
|
| 41 |
|
| 42 |
Cost for inference.
|
|
|
|
| 36 |
|
| 37 |
Dependencies
|
| 38 |
```
|
| 39 |
+
pip install -u torch[cuda] transformers einops
|
| 40 |
```
|
| 41 |
|
| 42 |
Cost for inference.
|