Update README.md
Browse files
README.md
CHANGED
|
@@ -182,8 +182,8 @@ Here provides a code snippet to show you how to load the tokenizer and model and
|
|
| 182 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 183 |
device = "cuda" # the device to load the model
|
| 184 |
|
| 185 |
-
model = AutoModelForCausalLM.from_pretrained("sail/Sailor-
|
| 186 |
-
tokenizer = AutoTokenizer.from_pretrained("sail/Sailor-
|
| 187 |
|
| 188 |
input_message = "Model bahasa adalah model probabilistik"
|
| 189 |
### The given Indonesian input translates to 'A language model is a probabilistic model of.'
|
|
|
|
| 182 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 183 |
device = "cuda" # the device to load the model
|
| 184 |
|
| 185 |
+
model = AutoModelForCausalLM.from_pretrained("sail/Sailor-4B", device_map="auto")
|
| 186 |
+
tokenizer = AutoTokenizer.from_pretrained("sail/Sailor-4B")
|
| 187 |
|
| 188 |
input_message = "Model bahasa adalah model probabilistik"
|
| 189 |
### The given Indonesian input translates to 'A language model is a probabilistic model of.'
|