Update README.md
Browse files
README.md
CHANGED
|
@@ -18,6 +18,7 @@ The model can be loaded as following:
|
|
| 18 |
|
| 19 |
```
|
| 20 |
import transformers
|
|
|
|
| 21 |
tokenizer = transformers.LlamaTokenizer.from_pretrained('chaoyi-wu/PMC_LLAMA_7B')
|
| 22 |
model = transformers.LlamaForCausalLM.from_pretrained('chaoyi-wu/PMC_LLAMA_7B')
|
| 23 |
sentence = 'Hello, doctor'
|
|
|
|
| 18 |
|
| 19 |
```
|
| 20 |
import transformers
|
| 21 |
+
import torch
|
| 22 |
tokenizer = transformers.LlamaTokenizer.from_pretrained('chaoyi-wu/PMC_LLAMA_7B')
|
| 23 |
model = transformers.LlamaForCausalLM.from_pretrained('chaoyi-wu/PMC_LLAMA_7B')
|
| 24 |
sentence = 'Hello, doctor'
|