Update README.md
Browse files
README.md
CHANGED
|
@@ -32,7 +32,7 @@ To load and run ChemDFM locally, here is an example:
|
|
| 32 |
|
| 33 |
```python
|
| 34 |
import torch
|
| 35 |
-
from
|
| 36 |
|
| 37 |
model_name_or_id = "OpenDFM/ChemDFM-v1.5-8B"
|
| 38 |
tokenizer = AutoTokenizer.from_pretrained(model_name_or_id)
|
|
|
|
| 32 |
|
| 33 |
```python
|
| 34 |
import torch
|
| 35 |
+
from modelscope import AutoTokenizer, LlamaForCausalLM, GenerationConfig
|
| 36 |
|
| 37 |
model_name_or_id = "OpenDFM/ChemDFM-v1.5-8B"
|
| 38 |
tokenizer = AutoTokenizer.from_pretrained(model_name_or_id)
|