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
|