Update README.md
Browse files
README.md
CHANGED
|
@@ -147,13 +147,13 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
| 147 |
import torch
|
| 148 |
|
| 149 |
repo = "kyujinpy/Sakura-SOLAR-Instruct"
|
| 150 |
-
|
| 151 |
repo,
|
| 152 |
return_dict=True,
|
| 153 |
torch_dtype=torch.float16,
|
| 154 |
device_map='auto'
|
| 155 |
)
|
| 156 |
-
|
| 157 |
```
|
| 158 |
|
| 159 |
---
|
|
|
|
| 147 |
import torch
|
| 148 |
|
| 149 |
repo = "kyujinpy/Sakura-SOLAR-Instruct"
|
| 150 |
+
sakura_solar = AutoModelForCausalLM.from_pretrained(
|
| 151 |
repo,
|
| 152 |
return_dict=True,
|
| 153 |
torch_dtype=torch.float16,
|
| 154 |
device_map='auto'
|
| 155 |
)
|
| 156 |
+
sakura_solar_tokenizer = AutoTokenizer.from_pretrained(repo)
|
| 157 |
```
|
| 158 |
|
| 159 |
---
|