Update README.md
Browse files
README.md
CHANGED
|
@@ -95,6 +95,7 @@ pip install -r requirements.txt
|
|
| 95 |
|
| 96 |
```python
|
| 97 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
|
|
| 98 |
|
| 99 |
model_name = "tokyotech-llm/Swallow-MS-7b-v0.1"
|
| 100 |
tokenizer = AutoTokenizer.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
|
|
|
|
| 95 |
|
| 96 |
```python
|
| 97 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 98 |
+
import torch
|
| 99 |
|
| 100 |
model_name = "tokyotech-llm/Swallow-MS-7b-v0.1"
|
| 101 |
tokenizer = AutoTokenizer.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
|