Update README.md
Browse files
README.md
CHANGED
@@ -10,6 +10,8 @@ language:
|
|
10 |
|
11 |
### Use
|
12 |
|
|
|
|
|
13 |
```python
|
14 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
15 |
|
|
|
10 |
|
11 |
### Use
|
12 |
|
13 |
+
requires: `transformers, einops`
|
14 |
+
|
15 |
```python
|
16 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
17 |
|