Commit
•
e3fcc0e
1
Parent(s):
a849304
Update README.md
Browse files
README.md
CHANGED
|
@@ -77,6 +77,7 @@ pip install -r requirements.txt
|
|
| 77 |
|
| 78 |
```python
|
| 79 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
|
|
| 80 |
|
| 81 |
model_name = "tokyotech-llm/Swallow-MX-8x7b-NVE-v0.1"
|
| 82 |
tokenizer = AutoTokenizer.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
|
|
|
|
| 77 |
|
| 78 |
```python
|
| 79 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 80 |
+
import torch
|
| 81 |
|
| 82 |
model_name = "tokyotech-llm/Swallow-MX-8x7b-NVE-v0.1"
|
| 83 |
tokenizer = AutoTokenizer.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
|