TravelLeraLone commited on
Commit
309861d
·
verified ·
1 Parent(s): 6e9a472

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 transformers import AutoTokenizer, LlamaForCausalLM, GenerationConfig
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)