Taishi-N324
commited on
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")
|