Duke-de-Artois
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -28,7 +28,15 @@ Codebase and datasets can be found at https://github.com/AI4Chem/ChemVlm.
|
|
28 |
|
29 |
|
30 |
Quick start as below(```transformers>=4.37.0 is needed```)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
|
|
|
32 |
```Python
|
33 |
from transformers import AutoTokenizer, AutoModelforCasualLM
|
34 |
import torch
|
|
|
28 |
|
29 |
|
30 |
Quick start as below(```transformers>=4.37.0 is needed```)
|
31 |
+
Update: You may also need
|
32 |
+
```
|
33 |
+
pip install sentencepiece
|
34 |
+
pip install einops
|
35 |
+
pip install timm
|
36 |
+
pip install accelerate>=0.26.0
|
37 |
+
```
|
38 |
|
39 |
+
Code:
|
40 |
```Python
|
41 |
from transformers import AutoTokenizer, AutoModelforCasualLM
|
42 |
import torch
|